Use the parameter End stillness

app.py

@@ -919,7 +919,7 @@ def worker_start_end(input_image, end_image, image_position, prompts, n_prompt,
 
 # 20250506 pftq: Modified worker to accept video input and clean frame count
 @torch.no_grad()
-def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
+def worker_video(input_video, end_frame, end_stillness, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
     def encode_prompt(prompt, n_prompt):
         llama_vec, clip_l_pooler = encode_prompt_conds(prompt, text_encoder, text_encoder_2, tokenizer, tokenizer_2)
 
@@ -994,7 +994,7 @@ def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolut
                 image_encoder=image_encoder, feature_extractor=feature_extractor, device=gpu
             )[:2]
             end_latent = end_latent.to(dtype=torch.float32, device=cpu)
-            end_latent = end_latent.expand(-1, -1, 100, -1, -1)
+            end_latent = end_latent.expand(-1, -1, 1 + end_stillness, -1, -1)
         else:
             end_latent = end_clip_embedding = None
 
@@ -1029,7 +1029,7 @@ def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolut
             def callback(d):
                 return
 
-        def compute_latent(history_latents, latent_window_size, latent_padding_size, num_clean_frames, start_latent, end_latent):
+        def compute_latent(history_latents, latent_window_size, latent_padding_size, num_clean_frames, start_latent, end_latent, end_stillness):
             # 20250506 pftq: Use user-specified number of context frames, matching original allocation for num_clean_frames=2
             available_frames = history_latents.shape[2]  # Number of latent frames
             max_pixel_frames = min(latent_window_size * 4 - 3, available_frames * 4)  # Cap at available pixel frames
@@ -1044,9 +1044,9 @@ def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolut
             total_context_frames = min(total_context_frames, available_frames)  # 20250507 pftq: Edge case for <=1 sec videos
 
             post_frames = 100 # Single frame for end_latent, otherwise padding causes still image
-            indices = torch.arange(0, 1 + num_4x_frames + num_2x_frames + effective_clean_frames + adjusted_latent_frames + ((latent_padding_size + post_frames) if end_latent is not None else 0)).unsqueeze(0) # 20250507 pftq: latent_window_size to adjusted_latent_frames for edge case for <=1 sec videos
+            indices = torch.arange(0, 1 + num_4x_frames + num_2x_frames + effective_clean_frames + adjusted_latent_frames + ((latent_padding_size + 1 + end_stillness) if end_latent is not None else 0)).unsqueeze(0) # 20250507 pftq: latent_window_size to adjusted_latent_frames for edge case for <=1 sec videos
             clean_latent_indices_start, clean_latent_4x_indices, clean_latent_2x_indices, clean_latent_1x_indices, latent_indices, blank_indices, clean_latent_indices_post = indices.split(
-                [1, num_4x_frames, num_2x_frames, effective_clean_frames, adjusted_latent_frames, latent_padding_size if end_latent is not None else 0, post_frames if end_latent is not None else 0], dim=1 # 20250507 pftq: latent_window_size to adjusted_latent_frames for edge case for <=1 sec videos
+                [1, num_4x_frames, num_2x_frames, effective_clean_frames, adjusted_latent_frames, latent_padding_size if end_latent is not None else 0, (1 + end_stillness) if end_latent is not None else 0], dim=1 # 20250507 pftq: latent_window_size to adjusted_latent_frames for edge case for <=1 sec videos
             )
             clean_latent_indices = torch.cat([clean_latent_indices_start, clean_latent_1x_indices, clean_latent_indices_post], dim=1)
 
@@ -1137,7 +1137,7 @@ def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolut
                 else:
                     transformer.initialize_teacache(enable_teacache=False)
 
-                [max_frames, clean_latents, clean_latents_2x, clean_latents_4x, latent_indices, clean_latents, clean_latent_indices, clean_latent_2x_indices, clean_latent_4x_indices] = compute_latent(history_latents, latent_window_size, latent_padding_size, num_clean_frames, start_latent, end_latent)
+                [max_frames, clean_latents, clean_latents_2x, clean_latents_4x, latent_indices, clean_latents, clean_latent_indices, clean_latent_2x_indices, clean_latent_4x_indices] = compute_latent(history_latents, latent_window_size, latent_padding_size, num_clean_frames, start_latent, end_latent, end_stillness)
 
                 generated_latents = sample_hunyuan(
                     transformer=transformer,
@@ -1348,7 +1348,7 @@ def process_video_on_gpu(input_video, end_frame, end_stillness, prompts, n_promp
     stream = AsyncStream()
 
     # 20250506 pftq: Pass num_clean_frames, vae_batch, etc
-    async_run(worker_video, input_video, end_frame, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
+    async_run(worker_video, input_video, end_frame, end_stillness, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
 
     output_filename = None