LORA params

app.py

@@ -811,7 +811,10 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
     stream.output_queue.push(('end', None))
     return
 
-def get_duration(input_image, image_position, prompts, generation_mode, n_prompt, seed, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, mp4_crf):
+def get_duration(input_image, image_position, prompts, generation_mode, n_prompt, seed, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, mp4_crf,
+            lora_file,
+            lora_multiplier,
+            fp8_optimization):
     global total_second_length_debug_value
 
     if total_second_length_debug_value is not None:
@@ -838,7 +841,10 @@ def process_on_gpu(input_image,
             gpu_memory_preservation=6,
             enable_preview=True,
             use_teacache=False,
-            mp4_crf=16
+            mp4_crf=16,
+            lora_file=None,
+            lora_multiplier=0.8,
+            fp8_optimization=False
            ):
     start = time.time()
     global stream
@@ -936,10 +942,16 @@ def process(input_image,
             gpu_memory_preservation,
             enable_preview,
             use_teacache,
-            mp4_crf
+            mp4_crf,
+            lora_file,
+            lora_multiplier,
+            fp8_optimization
            )
 
-def get_duration_video(input_video, 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 get_duration_video(input_video, 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,
+            lora_file,
+            lora_multiplier,
+            fp8_optimization):
     global total_second_length_debug_value
     if total_second_length_debug_value is not None:
         duration = min(total_second_length_debug_value * 60 * 10, 600)
@@ -949,7 +961,10 @@ def get_duration_video(input_video, prompts, n_prompt, seed, batch, resolution,
 
 # Remove this decorator if you run on local
 @spaces.GPU(duration=get_duration_video)
-def process_video_on_gpu(input_video, 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 process_video_on_gpu(input_video, 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,
+            lora_file,
+            lora_multiplier,
+            fp8_optimization):
     start = time.time()
     global stream
     stream = AsyncStream()
@@ -1021,7 +1036,10 @@ def process_video(input_video, prompt, n_prompt, randomize_seed, seed, batch, re
     if cfg > 1:
         gs = 1
 
-    yield from process_video_on_gpu(input_video, 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)
+    yield from process_video_on_gpu(input_video, 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,
+            lora_file,
+            lora_multiplier,
+            fp8_optimization)
 
 def end_process():
     stream.input_queue.push('end')