fffiloni commited on
Commit
927c90e
·
verified ·
1 Parent(s): 26557da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -10,7 +10,18 @@ from wan_loader import load_wan_pipe
10
  from models.set_condition_branch import set_stand_in
11
  from preprocessor import FaceProcessor
12
 
 
 
13
  print("Loading model, please wait...")
 
 
 
 
 
 
 
 
 
14
  try:
15
  ANTELOPEV2_PATH = "checkpoints/antelopev2"
16
  BASE_MODEL_PATH = "checkpoints/base_model/"
 
10
  from models.set_condition_branch import set_stand_in
11
  from preprocessor import FaceProcessor
12
 
13
+ from huggingface_hub import snapshot_download
14
+
15
  print("Loading model, please wait...")
16
+
17
+ snapshot_download("Wan-AI/Wan2.1-T2V-14B", local_dir="checkpoints/base_model/")
18
+
19
+ snapshot_download(
20
+ "DIAMONIK7777/antelopev2",
21
+ local_dir="checkpoints/antelopev2/models/antelopev2"
22
+ )
23
+ snapshot_download("BowenXue/Stand-In", local_dir="checkpoints/Stand-In/")
24
+
25
  try:
26
  ANTELOPEV2_PATH = "checkpoints/antelopev2"
27
  BASE_MODEL_PATH = "checkpoints/base_model/"