ICGenAIShare04 commited on
Commit
a10589c
·
verified ·
1 Parent(s): e31bd94

Update src/video_generator_hf.py

Browse files

fix: update torchao FP8 API and reduce video inference steps to 25

Files changed (1) hide show
  1. src/video_generator_hf.py +1 -1
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 = 30
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