inoculatemedia commited on
Commit
cfa4385
·
verified ·
1 Parent(s): 1412ed3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -43,6 +43,14 @@ from einops import rearrange
43
  import soundfile as sf
44
  import re
45
 
 
 
 
 
 
 
 
 
46
  def _validate_args(args):
47
  # Basic check
48
  assert args.ckpt_dir is not None, "Please specify the checkpoint directory."
 
43
  import soundfile as sf
44
  import re
45
 
46
+ from huggingface_hub import hf_hub_download
47
+
48
+ checkpoint_path = hf_hub_download(
49
+ repo_id="Wan-AI/Wan2.1-I2V-14B-480P",
50
+ filename="models_t5_umt5-xxl-enc-bf16.pth",
51
+ repo_type="model"
52
+ )
53
+
54
  def _validate_args(args):
55
  # Basic check
56
  assert args.ckpt_dir is not None, "Please specify the checkpoint directory."