Yuanshi commited on
Commit
f2b3d29
·
1 Parent(s): 63581e5
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -24,7 +24,10 @@ def get_fps(path):
24
 
25
  base_model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
26
  pipe = WanPipeline.from_pretrained(
27
- base_model_id, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True
 
 
 
28
  ).to("cuda")
29
  load_vibt_weight(
30
  pipe.transformer,
 
24
 
25
  base_model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
26
  pipe = WanPipeline.from_pretrained(
27
+ base_model_id,
28
+ torch_dtype=torch.bfloat16,
29
+ low_cpu_mem_usage=True,
30
+ keep_in_fp32_modules=False, # <-- 或者你也可以强制关掉
31
  ).to("cuda")
32
  load_vibt_weight(
33
  pipe.transformer,