Update src/video_generator_hf.py
Browse filesfix: update torchao FP8 API and reduce video inference steps to 25
src/video_generator_hf.py
CHANGED
|
@@ -21,7 +21,7 @@ MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
|
|
| 21 |
|
| 22 |
NUM_FRAMES = 81 # ~5s at 16fps
|
| 23 |
FPS = 16
|
| 24 |
-
NUM_INFERENCE_STEPS =
|
| 25 |
GUIDANCE_SCALE = 5.0
|
| 26 |
SEED = 42
|
| 27 |
|
|
|
|
| 21 |
|
| 22 |
NUM_FRAMES = 81 # ~5s at 16fps
|
| 23 |
FPS = 16
|
| 24 |
+
NUM_INFERENCE_STEPS = 25
|
| 25 |
GUIDANCE_SCALE = 5.0
|
| 26 |
SEED = 42
|
| 27 |
|