jbilcke-hf HF staff commited on
Commit
9a3f2e6
·
verified ·
1 Parent(s): 1dee97c

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -296,7 +296,7 @@ class EndpointHandler:
296
  - enable_teacache (optional, bool, default to True): Generate faster at the cost of a slight quality loss
297
  - teacache_threshold (optional, float, default to 0.05): Amount of cache, 0 (original), 0.03 (1.6x speedup), 0.05 (Default, 2.1x speedup).
298
  - enable_enhance_a_video (optional, bool, default to True): enable the enhance_a_video optimization
299
- - enhance_a_video_weight(optional, float, default to 4.0): amount of video enhancement to apply
300
  - lora_model_name(optional, str, default to ""): HuggingFace repo ID or path to LoRA model
301
  - lora_model_weight_file(optional, str, default to ""): Specific weight file to load from the LoRA model
302
  - lora_model_trigger(optional, str, default to ""): Optional trigger word to prepend to the prompt
@@ -355,7 +355,7 @@ class EndpointHandler:
355
 
356
  # Add enhance-a-video settings
357
  enable_enhance_a_video=params.get("enable_enhance_a_video", True),
358
- enhance_a_video_weight=params.get("enhance_a_video_weight", 4.0),
359
 
360
  # LoRA settings
361
  lora_model_name=params.get("lora_model_name", ""),
 
296
  - enable_teacache (optional, bool, default to True): Generate faster at the cost of a slight quality loss
297
  - teacache_threshold (optional, float, default to 0.05): Amount of cache, 0 (original), 0.03 (1.6x speedup), 0.05 (Default, 2.1x speedup).
298
  - enable_enhance_a_video (optional, bool, default to True): enable the enhance_a_video optimization
299
+ - enhance_a_video_weight(optional, float, default to 5.0): amount of video enhancement to apply
300
  - lora_model_name(optional, str, default to ""): HuggingFace repo ID or path to LoRA model
301
  - lora_model_weight_file(optional, str, default to ""): Specific weight file to load from the LoRA model
302
  - lora_model_trigger(optional, str, default to ""): Optional trigger word to prepend to the prompt
 
355
 
356
  # Add enhance-a-video settings
357
  enable_enhance_a_video=params.get("enable_enhance_a_video", True),
358
+ enhance_a_video_weight=params.get("enhance_a_video_weight", 5.0),
359
 
360
  # LoRA settings
361
  lora_model_name=params.get("lora_model_name", ""),