total_second_length

app_endframe.py

@@ -64,7 +64,7 @@ vae = AutoencoderKLHunyuanVideo.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo"
 feature_extractor = SiglipImageProcessor.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='feature_extractor')
 image_encoder = SiglipVisionModel.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='image_encoder', torch_dtype=torch.float16).cpu()
 
-transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('Fabrice-TIERCELIN/FramePackI2V_HY', torch_dtype=torch.bfloat16).cpu()
+transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('lllyasviel/FramePackI2V_HY', torch_dtype=torch.bfloat16).cpu()
 
 vae.eval()
 text_encoder.eval()
@@ -108,7 +108,8 @@ outputs_folder = './outputs/'
 os.makedirs(outputs_folder, exist_ok=True)
 
 input_video_debug_value = None
-prompt_debug_value = None
+prompt_debug_valuetotal_second_length_debug_value
+total_second_length_debug_value
 
 # 20250506 pftq: Added function to encode input video frames into latents
 @torch.no_grad()
@@ -666,11 +667,14 @@ def worker(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, bat
     
 # 20250506 pftq: Modified process to pass clean frame count, etc
 def get_duration(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
+    global total_second_length_debug_value
+    if total_second_length_debug_value is not None:
+        return total_second_length_debug_value * 60
     return total_second_length * 60
 
 @spaces.GPU(duration=get_duration)
 def process(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
-    global stream, high_vram, input_video_debug_value, prompt_debug_value
+    global stream, high_vram, input_video_debug_value, prompt_debug_value, total_second_length_debug_value
 
     if input_video_debug_value is not None:
         input_video = input_video_debug_value
@@ -680,6 +684,10 @@ def process(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, ba
         prompt = prompt_debug_value
         prompt_debug_value = None
 
+    if total_second_length_debug_value is not None:
+        total_second_length = total_second_length_debug_value
+        total_second_length_debug_value = None
+
     # 20250506 pftq: Updated assertion for video input
     assert input_video is not None, 'No input video!'
 
@@ -798,6 +806,7 @@ with block:
             with gr.Row():
                 input_video_debug = gr.Video(sources='upload', label="Input Video Debug", height=320)
                 prompt_debug = gr.Textbox(label="Prompt Debug", value='')
+                total_second_length_debug = gr.Slider(label="Additional Video Length to Generate (Seconds) Debug", minimum=1, maximum=120, value=5, step=0.1)
 
         with gr.Column():
             preview_image = gr.Image(label="Next Latents", height=200, visible=False)
@@ -857,6 +866,11 @@ with block:
         prompt_debug_value = input
         return []
     
+    def handle_total_second_length_debug_change(input):
+        global total_second_length_debug_value
+        total_second_length_debug_value = input
+        return []
+    
     input_video_debug.upload(
         fn=handle_input_video_debug_upload,
         inputs=[input_video_debug],
@@ -868,5 +882,11 @@ with block:
         inputs=[prompt_debug],
         outputs=[]
     )
+    
+    total_second_length_debug.change(
+        fn=handle_total_second_length_debug_change,
+        inputs=[total_second_length_debug],
+        outputs=[]
+    )
 
 block.launch(share=True)