Update app.py

app.py

@@ -208,17 +208,29 @@ def generate_multi_view_images(prompt, seed):
     # generator = torch.manual_seed(seed)
     generator = torch.Generator().manual_seed(seed)
     with torch.no_grad():
-        images = flux_pipe(
+        # images = flux_pipe(
+        #     prompt=prompt,
+        #     num_inference_steps=10,
+        #     guidance_scale=3.5,
+        #     num_images_per_prompt=1,
+        #     width=resolution * 4,
+        #     height=resolution * 2,
+        #     output_type='np',
+        #     generator=generator,
+        #     good_vae=good_vae,
+        # ).images
+        for images in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt,
-            num_inference_steps=10,
             guidance_scale=3.5,
-            num_images_per_prompt=1,
+            num_inference_steps=20,
             width=resolution * 4,
-            height=resolution * 2,
-            output_type='np',
+            height==resolution * 2,
             generator=generator,
+            output_type="np",
             good_vae=good_vae,
-        ).images
+        ):
+        pass
+    # 返回最终的图像和种子（通过外部调用处理）
     return images
 
 # 重建 3D 模型