Spaces:

rahul7star
/

Wan2.1-Text-Vdo-Demp

Running

rahul7star commited on Mar 5

Commit

545fccc

verified ·

1 Parent(s): fb7dcce

Update generate.py

Files changed (1) hide show

generate.py CHANGED Viewed

@@ -14,7 +14,7 @@ parser.add_argument("--ckpt_dir", type=str, default="./Wan2.1-T2V-14B")
 parser.add_argument("--offload_model", type=str, default="True")
 parser.add_argument("--prompt", type=str, required=True)
 args = parser.parse_args()
- print("🔄 Downloading WAN 2.1 start..")
 # Ensure the model is downloaded
 if not os.path.exists(args.ckpt_dir):
     print("🔄 Downloading WAN 2.1 - 14B model from Hugging Face...")

 parser.add_argument("--offload_model", type=str, default="True")
 parser.add_argument("--prompt", type=str, required=True)
 args = parser.parse_args()
+print("🔄 Downloading WAN 2.1 start..")
 # Ensure the model is downloaded
 if not os.path.exists(args.ckpt_dir):
     print("🔄 Downloading WAN 2.1 - 14B model from Hugging Face...")