Update handler.py
Browse files- handler.py +4 -1
handler.py
CHANGED
@@ -45,7 +45,7 @@ class GenerationConfig:
|
|
45 |
|
46 |
# TeaCache settings
|
47 |
enable_teacache: bool = True
|
48 |
-
teacache_threshold: float = 0.15
|
49 |
|
50 |
|
51 |
# Enhance-A-Video settings
|
@@ -139,7 +139,10 @@ class EndpointHandler:
|
|
139 |
audio_prompt=params.get("audio_prompt", ""),
|
140 |
audio_negative_prompt=params.get("audio_negative_prompt", "voices, voice, talking, speaking, speech"),
|
141 |
enable_teacache=params.get("enable_teacache", True),
|
|
|
|
|
142 |
teacache_threshold=params.get("teacache_threshold", 0.15),
|
|
|
143 |
enable_enhance_a_video=params.get("enable_enhance_a_video", True),
|
144 |
enhance_a_video_weight=params.get("enhance_a_video_weight", 4.0)
|
145 |
).validate_and_adjust()
|
|
|
45 |
|
46 |
# TeaCache settings
|
47 |
enable_teacache: bool = True
|
48 |
+
teacache_threshold: float = 0.15 # values: 0 (original), 0.1 (1.6x speedup), 0.15 (2.1x speedup)
|
49 |
|
50 |
|
51 |
# Enhance-A-Video settings
|
|
|
139 |
audio_prompt=params.get("audio_prompt", ""),
|
140 |
audio_negative_prompt=params.get("audio_negative_prompt", "voices, voice, talking, speaking, speech"),
|
141 |
enable_teacache=params.get("enable_teacache", True),
|
142 |
+
|
143 |
+
# values: 0 (original), 0.1 (1.6x speedup), 0.15 (2.1x speedup).
|
144 |
teacache_threshold=params.get("teacache_threshold", 0.15),
|
145 |
+
|
146 |
enable_enhance_a_video=params.get("enable_enhance_a_video", True),
|
147 |
enhance_a_video_weight=params.get("enhance_a_video_weight", 4.0)
|
148 |
).validate_and_adjust()
|