Update handler.py
Browse files- 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
|
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",
|
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", ""),
|