Hugging Face's logo

Spaces:
heboya8
/
Wan2.1-T2V-1.3B-Local
Paused

App Files Community
SeedOfEvil commited on
Commit
8ab574b
·
verified ·
1 Parent(s): 8d4efd3

Update download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +0 -1
download_model.py CHANGED
@@ -3,7 +3,6 @@ from diffusers import AutoencoderKLWan, WanPipeline
3
 
4
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
5
  print("Pre-downloading model weights on CPU...")
6
- # Load the VAE and pipeline on CPU only
7
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
8
  _ = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
9
  print("Model weights pre-downloaded.")
 
3
 
4
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
5
  print("Pre-downloading model weights on CPU...")
 
6
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
7
  _ = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
8
  print("Model weights pre-downloaded.")