Allocation time

app.py

@@ -1099,6 +1099,7 @@ class CameraDropdown(gr.HTML):
         )
 
 def generate_video_example_debug(input_image, prompt, camera_lora, resolution, radioanimated_mode, input_video, input_audio, progress=gr.Progress(track_tqdm=True)):
+    allocation_time = None
 
     if input_image_debug_value[0] is not None or input_audio_debug_value[0] is not None or input_video_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None or allocation_time_debug_value[0] is not None or resolution_debug_value[0] is not None or factor_debug_value[0] is not None:
         input_image = input_image_debug_value[0]
@@ -1108,9 +1109,9 @@ def generate_video_example_debug(input_image, prompt, camera_lora, resolution, r
         duration = total_second_length_debug_value[0]
         resolution = resolution_debug_value[0]
         allocation_time = allocation_time_debug_value[0]
-    return generate_video_example(input_image, prompt, camera_lora, resolution, radioanimated_mode, input_video, input_audio)
+    return generate_video_example(input_image, prompt, camera_lora, resolution, radioanimated_mode, input_video, input_audio, allocation_time)
 
-def generate_video_example(input_image, prompt, camera_lora, resolution, radioanimated_mode, input_video, input_audio, progress=gr.Progress(track_tqdm=True)):
+def generate_video_example(input_image, prompt, camera_lora, resolution, radioanimated_mode, input_video, input_audio, allocation_time=None, progress=gr.Progress(track_tqdm=True)):
 
     w, h = apply_resolution(resolution)
 
@@ -1128,6 +1129,7 @@ def generate_video_example(input_image, prompt, camera_lora, resolution, radioan
             w,   
             camera_lora,     
             input_audio,
+            allocation_time,
             progress
         )
     return output_video
@@ -1145,8 +1147,12 @@ def get_duration(
     width,
     camera_lora,
     audio_path,
+    allocation_time,
     progress
 ):
+    if allocation_time is not None:
+        allocation_time
+
     extra_time = 0
 
     if audio_path is not None:
@@ -1179,6 +1185,7 @@ def generate_video(
     width: int = DEFAULT_1_STAGE_WIDTH,
     camera_lora: str = "No LoRA", 
     audio_path = None,
+    allocation_time = None,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -2347,7 +2354,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
         input_image_debug=gr.Image(type="filepath", label="Image Debug")
         input_audio_debug = gr.Audio(label="Audio Debug", type="filepath")
         input_video_debug=gr.Video(label="Video Debug")
-        total_second_length_debug=gr.Slider(label="Duration Debug", minimum=1, maximum=120, value=5, step=0.1)
+        total_second_length_debug=gr.Slider(label="Duration Debug", minimum=1, maximum=120, value=7, step=0.1)
         resolution_debug = gr.Dropdown(
                             choices=[
                                 ["16:9", "16:9"],
@@ -2358,7 +2365,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                             elem_id="resolution_ui",
             label="Resolution Debug")
         factor_debug=gr.Slider(label="Factor Debug", minimum=1, maximum=100, value=3.2, step=0.1)
-        allocation_time_debug=gr.Slider(label="Allocation Debug", minimum=1, maximum=60 * 20, value=720, step=1)
+        allocation_time_debug=gr.Slider(label="Allocation Debug", minimum=1, maximum=60 * 20, value=600, step=1)
 
     def handle_field_debug_change(
         input_image_debug_data,