Update model.py

model.py

@@ -266,7 +266,7 @@ class Model:
         samples = 255 * (samples * 0.5 + 0.5)
         samples = [image] + [
             Image.fromarray(
-                sample.permute(1, 2, 0).numpy().clip(0, 255).astype(np.uint8))
+                sample.permute(1, 2, 0).cpu().detach().numpy().clip(0, 255).astype(np.uint8))
             for sample in samples
         ]
         del image_tensor