Update pipeline.py

pipeline.py

@@ -1600,8 +1600,8 @@ class LGMFullPipeline(DiffusionPipeline):
         if hasattr(self, "final_offload_hook") and self.final_offload_hook is not None:
             self.final_offload_hook.offload()
 
-        # images = np.stack([image[1], image[2], image[3], image[0]], axis=0)
-        images = np.stack([x for x in image[1:]] + [image[0]], axis=0)
+        images = np.stack([image[1], image[2], image[3], image[0]], axis=0)
+        # images = np.stack([x for x in image[1:]] + [image[0]], axis=0)
         images = torch.from_numpy(images).permute(0, 3, 1, 2).float().cuda()
         images = F.interpolate(
             images,