jbilcke-hf HF staff commited on
Commit
8b8ed09
·
verified ·
1 Parent(s): 97a4a3d

Update handler.py

Browse files
Files changed (1) hide show
  1. 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()