rahul7star commited on
Commit
1b11e16
·
verified ·
1 Parent(s): 8d6b484

Update app_fast.py

Browse files
Files changed (1) hide show
  1. app_fast.py +1 -2
app_fast.py CHANGED
@@ -9,8 +9,7 @@ import numpy as np
9
  from PIL import Image
10
  import random
11
 
12
-
13
- MODEL_ID = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
14
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
15
 
16
  # Initialize pipelines
 
9
  from PIL import Image
10
  import random
11
 
12
+ MODEL_ID = "FastVideo/FastWan2.2-TI2V-5B-FullAttn-Diffusers"
 
13
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
14
 
15
  # Initialize pipelines