Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import sys
|
|
3 |
import subprocess
|
4 |
from huggingface_hub import hf_hub_download, hf_hub_list
|
5 |
from pathlib import Path
|
6 |
-
|
7 |
MODEL_REPO = "tencent/HunyuanVideo-Avatar"
|
8 |
BASE_DIR = Path.cwd()
|
9 |
WEIGHTS_DIR = BASE_DIR / "weights"
|
@@ -45,7 +45,8 @@ def download_ckpts(files):
|
|
45 |
else:
|
46 |
print(f"File {local_path} already exists, skipping download.")
|
47 |
print("✅ All checkpoint files downloaded.")
|
48 |
-
|
|
|
49 |
def run_sample_gpu_poor():
|
50 |
checkpoint_fp8 = CKPTS_DIR / "hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8.pt"
|
51 |
if not checkpoint_fp8.is_file():
|
|
|
3 |
import subprocess
|
4 |
from huggingface_hub import hf_hub_download, hf_hub_list
|
5 |
from pathlib import Path
|
6 |
+
import spaces
|
7 |
MODEL_REPO = "tencent/HunyuanVideo-Avatar"
|
8 |
BASE_DIR = Path.cwd()
|
9 |
WEIGHTS_DIR = BASE_DIR / "weights"
|
|
|
45 |
else:
|
46 |
print(f"File {local_path} already exists, skipping download.")
|
47 |
print("✅ All checkpoint files downloaded.")
|
48 |
+
|
49 |
+
@spaces.GPU(duration=1000)
|
50 |
def run_sample_gpu_poor():
|
51 |
checkpoint_fp8 = CKPTS_DIR / "hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8.pt"
|
52 |
if not checkpoint_fp8.is_file():
|