add duration

app.py

@@ -12,6 +12,9 @@ from app_utils import *
 from axis_renderer import BlendRenderer
 import spaces
 
+torch.backends.cuda.matmul.allow_tf32 = False
+torch.backends.cudnn.allow_tf32 = False
+
 from huggingface_hub import hf_hub_download
 ckpt_path = hf_hub_download(repo_id=ORIANY_V2, filename=REMOTE_CKPT_PATH, repo_type="model", cache_dir='./', resume_download=True)
 print(ckpt_path)
@@ -32,7 +35,7 @@ print('Model loaded.')
 
 axis_renderer = BlendRenderer(RENDER_FILE)
 
-@spaces.GPU
+@spaces.GPU(duration=20)
 @torch.no_grad()
 def inf_single_batch(batch):
     model.to(device='cuda', dtype=mark_dtype)
@@ -110,7 +113,7 @@ def safe_image_input(image):
 
 
 # ====== 推理函数 ======
-@spaces.GPU
+@spaces.GPU(duration=20)
 @torch.no_grad()
 def run_inference(image_ref, image_tgt, do_rm_bkg):
     image_ref = safe_image_input(image_ref)