Unload models for video extension

app.py

@@ -645,6 +645,11 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
             load_model_as_complete(image_encoder, target_device=gpu)
 
         image_encoder_output = hf_clip_vision_encode(input_image_np, feature_extractor, image_encoder)
+
+        # Clean GPU
+        if not high_vram:
+            unload_complete_models(image_encoder)
+
         image_encoder_last_hidden_state = image_encoder_output.last_hidden_state
 
         # Dtype
@@ -817,7 +822,7 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
                     history_pixels = soft_append_bcthw(history_pixels, vae_decode(real_history_latents, vae).cpu(), overlapped_frames)
 
                 if not high_vram:
-                    unload_complete_models()
+                    unload_complete_models(text_encoder, text_encoder_2, image_encoder, vae, transformer)
 
                 if enable_preview or section_index == total_latent_sections - 1:
                     output_filename = os.path.join(outputs_folder, f'{job_id}_{total_generated_latent_frames}.mp4')