upd

app.py

@@ -39,7 +39,7 @@ def check_video_input(image_block: Image.Image):
         raise gr.Error("Please generate a video first")
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU()
 def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider: int):
     if not os.path.exists('tmp_data'):
         os.makedirs('tmp_data')
@@ -60,7 +60,7 @@ def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider
     # return [os.path.join('logs', 'tmp_rgba_model.ply')]
     return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=200)
 def optimize_stage_2(image_block: Image.Image, seed_slider: int):
     img_hash = hashlib.sha256(image_block.tobytes()).hexdigest()
 

configs/4d_demo.yaml

@@ -30,7 +30,7 @@ lambda_svd: 0
 # training batch size per iter
 batch_size: 7
 # training iterations for stage 1
-iters: 300
+iters: 500
 # training iterations for stage 2
 iters_refine: 50
 # training camera radius