SeedOfEvil commited on
Commit
085959f
·
verified ·
1 Parent(s): 15e5927

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ def generate_video(prompt, negative_prompt=""):
10
 
11
  # Load the VAE and pipeline with proper data types per model requirements
12
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 
13
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
14
  pipe.to("cuda")
15
 
 
10
 
11
  # Load the VAE and pipeline with proper data types per model requirements
12
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
13
+ # With accelerate installed, low_cpu_mem_usage can now be enabled automatically.
14
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
15
  pipe.to("cuda")
16