Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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/"
|