Update app.py

app.py

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