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