rahul7star commited on
Commit
51c2eac
·
verified ·
1 Parent(s): 9877f80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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():