SeedOfEvil commited on
Commit
2d39a8b
·
verified ·
1 Parent(s): d178d24

Update download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +1 -1
download_model.py CHANGED
@@ -5,4 +5,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.")
 
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.")