linoyts HF Staff commited on
Commit
d4fdc43
·
verified ·
1 Parent(s): 654762e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
31
  MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
32
 
33
 
34
- pipe = WanTextToVideoPipeline.from_pretrained(MODEL_ID,
35
  transformer=WanTransformer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
36
  subfolder='transformer',
37
  torch_dtype=torch.bfloat16,
 
31
  MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
32
 
33
 
34
+ pipe = WanPipeline.from_pretrained(MODEL_ID,
35
  transformer=WanTransformer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
36
  subfolder='transformer',
37
  torch_dtype=torch.bfloat16,