contigousness

hydra_node/models.py

@@ -123,6 +123,7 @@ class StableDiffusionModel(nn.Module):
         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)
+            x_sample = np.ascontiguousarray(x_sample)
             images.append(x_sample) 
 
         return images
\ No newline at end of file