bug fix

gradio_app/gradio_3dgen.py

@@ -11,14 +11,14 @@ from scripts.utils import save_glb_and_video
 # from scripts.multiview_inference import geo_reconstruct
 from scripts.multiview_inference import geo_reconstruct_part1, geo_reconstruct_part2, geo_reconstruct_part3
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 def run_mv(preview_img, input_processing, seed):
     if preview_img.size[0] <= 512:
         preview_img = run_sr_fast([preview_img])[0]
     rgb_pils, front_pil = run_mvprediction(preview_img, remove_bg=input_processing, seed=int(seed)) # 6s
     return rgb_pils, front_pil
 
-# @spaces.GPU(duration=90) # seems split into multiple part will leads to `RuntimeError`, before fix it, still initialize here
+@spaces.GPU(duration=100) # seems split into multiple part will leads to `RuntimeError`, before fix it, still initialize here
 def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"):
     if preview_img is None:
         raise gr.Error("The input image is none!")

scripts/multiview_inference.py

@@ -104,7 +104,7 @@ def geo_reconstruct(rgb_pils, normal_pils, front_pil, do_refine=False, predict_n
 ########################
 import spaces
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 def geo_reconstruct_part1(rgb_pils, normal_pils, front_pil, do_refine=False, predict_normal=True, expansion_weight=0.1, init_type="std"):
     if front_pil.size[0] <= 512:
         front_pil = run_sr_fast([front_pil])[0]
@@ -149,7 +149,7 @@ def geo_reconstruct_part2(vertices, faces):
     meshes = simple_clean_mesh(to_pyml_mesh(vertices, faces), apply_smooth=True, stepsmoothnum=1, apply_sub_divide=True, sub_divide_threshold=0.25)
     return meshes
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 def geo_reconstruct_part3(meshes, img_list):
     meshes = meshes.to("cuda")
     new_meshes = multiview_color_projection(meshes, img_list, resolution=1024, device="cuda", complete_unseen=True, confidence_threshold=0.2, cameras_list = get_cameras_list([0, 90, 180, 270], "cuda", focal=1))