Update app.py

app.py

@@ -83,7 +83,7 @@ from huggingface_hub import hf_hub_download
 from utils.tool import NormalTransfer, get_background, get_render_cameras_video, load_mipmap, render_frames
 
 device_0 = "cuda:0"
-device_1 = "cuda:1"
+device_1 = "cuda:0"
 resolution = 512
 save_dir = "./outputs"
 normal_transfer = NormalTransfer()
@@ -203,7 +203,7 @@ def generate_multi_view_images(prompt, seed):
     with torch.no_grad():
         images = flux_pipe(
             prompt=prompt,
-            num_inference_steps=20,
+            num_inference_steps=15,
             guidance_scale=3.5,
             num_images_per_prompt=1,
             width=resolution * 4,