Yuanshi commited on
Commit
74b567f
·
1 Parent(s): 160287b
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -23,10 +23,10 @@ def get_fps(path):
23
 
24
 
25
  base_model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
26
- vae = AutoencoderKLWan.from_pretrained(
27
- base_model_id, subfolder="vae", torch_dtype=torch.float32
28
- )
29
- pipe = WanPipeline.from_pretrained(base_model_id, vae=vae, torch_dtype=torch.bfloat16)
30
  pipe.to("cuda")
31
  load_vibt_weight(
32
  pipe.transformer,
 
23
 
24
 
25
  base_model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
26
+ # vae = AutoencoderKLWan.from_pretrained(
27
+ # base_model_id, subfolder="vae", torch_dtype=torch.float32
28
+ # )
29
+ pipe = WanPipeline.from_pretrained(base_model_id, torch_dtype=torch.bfloat16)
30
  pipe.to("cuda")
31
  load_vibt_weight(
32
  pipe.transformer,