rahul7star commited on
Commit
956b411
·
verified ·
1 Parent(s): 093967b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,8 +4,9 @@ import subprocess
4
  import time
5
 
6
  MODEL_REPO = "tencent/HunyuanVideo-Avatar"
7
- BASE_DIR = "HunyuanVideo-Avatar"
8
  WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
 
9
  CHECKPOINT_FILE = os.path.join(WEIGHTS_DIR, "transformers/mp_rank_00_model_states.pt") # Adjust if needed
10
 
11
  def download_model():
 
4
  import time
5
 
6
  MODEL_REPO = "tencent/HunyuanVideo-Avatar"
7
+ BASE_DIR = os.getcwd() # current working directory, absolute path
8
  WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
9
+
10
  CHECKPOINT_FILE = os.path.join(WEIGHTS_DIR, "transformers/mp_rank_00_model_states.pt") # Adjust if needed
11
 
12
  def download_model():