Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|