4 seconds

app.py

@@ -127,7 +127,6 @@ def calculate_required_time(steps, max_duration):
         2: 8,
         3: 11,
         4: 20,
-        5: 30,
     }
     each_step_s = max_duration_duration_mapping[max_duration]
     duration_s = (each_step_s * steps) + warmup_s
@@ -161,7 +160,7 @@ def generate_scene(prompt_text, steps, image_paths, audio_file_path, tea_cache_l
     return run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration, session_id, progress)
 
 
-def generate_example(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 5, session_id = None, progress=gr.Progress(track_tqdm=True)):
+def generate_example(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 4, session_id = None, progress=gr.Progress(track_tqdm=True)):
 
     return run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration, session_id, progress)
 
@@ -257,8 +256,7 @@ def run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_
         1:25,
         2:45,
         3:70, 
-        4:97,
-        5:129
+        4:97
     }
 
     # Run inference