as uint8

hydra_node/models.py

@@ -122,6 +122,7 @@ class StableDiffusionModel(nn.Module):
         images = []
         for x_sample in x_samples_ddim:
             x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
+            x_sample = x_sample.astype(np.uint8)
             images.append(x_sample) 
 
         return images
\ No newline at end of file