detach unnecessary

2022-07-08 13:26:53 -04:00
parent 1b45a5de2a
commit 703bfb231d
2 changed files with 3 additions and 3 deletions
--- a/min_dalle/min_dalle.py
+++ b/min_dalle/min_dalle.py
@@ -161,7 +161,7 @@ class MinDalle:
        grid_size = int(sqrt(images.shape[0]))
        images = images.reshape([grid_size] * 2 + list(images.shape[1:]))
        image = images.flatten(1, 2).transpose(0, 1).flatten(1, 2)
-        image = Image.fromarray(image.to('cpu').detach().numpy())
+        image = Image.fromarray(image.to('cpu').numpy())
        return image