Bils commited on
Commit
0fc14ac
·
verified ·
1 Parent(s): 4c81b6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -94,7 +94,7 @@ def prepare_once():
94
  """Clone code (skip LFS), download weights, set env, prepare dirs."""
95
  _clone_without_lfs()
96
 
97
- # Ensure we can import their package
98
  if str(REPO_DIR) not in sys.path:
99
  sys.path.insert(0, str(REPO_DIR))
100
 
@@ -113,9 +113,9 @@ def prepare_once():
113
 
114
  prepare_once()
115
 
116
- # ---- Friendly dependency check (correct package name) ------------------------
117
  try:
118
- import audiotools # provided by the PyPI package 'descript-audiotools'
119
  except Exception as e:
120
  raise RuntimeError(
121
  "Missing module 'audiotools'. Install it via the PyPI package "
@@ -123,7 +123,17 @@ except Exception as e:
123
  "to requirements.txt) and restart the Space."
124
  ) from e
125
 
126
- # Now safe to import their internals
 
 
 
 
 
 
 
 
 
 
127
  from hunyuanvideo_foley.utils.model_utils import load_model, denoise_process
128
  from hunyuanvideo_foley.utils.feature_utils import feature_process
129
  from hunyuanvideo_foley.utils.media_utils import merge_audio_video
@@ -153,7 +163,7 @@ def _setup_device(device_str: str = "auto", gpu_id: int = 0) -> torch.device:
153
  return d
154
 
155
  def auto_load_models() -> str:
156
- """Download weights if needed + load model natively."""
157
  global _model_dict, _cfg, _device
158
 
159
  if not os.path.exists(MODEL_PATH):
 
94
  """Clone code (skip LFS), download weights, set env, prepare dirs."""
95
  _clone_without_lfs()
96
 
97
+ # Ensure we can import their package later
98
  if str(REPO_DIR) not in sys.path:
99
  sys.path.insert(0, str(REPO_DIR))
100
 
 
113
 
114
  prepare_once()
115
 
116
+
117
  try:
118
+ import audiotools
119
  except Exception as e:
120
  raise RuntimeError(
121
  "Missing module 'audiotools'. Install it via the PyPI package "
 
123
  "to requirements.txt) and restart the Space."
124
  ) from e
125
 
126
+ try:
127
+ import omegaconf # noqa: F401
128
+ import yaml # from pyyaml
129
+ import easydict # noqa: F401
130
+ except Exception as e:
131
+ raise RuntimeError(
132
+ "Missing config deps. Please add to requirements.txt: "
133
+ "'omegaconf>=2.3.0', 'pyyaml', and 'easydict'."
134
+ ) from e
135
+
136
+ # Now safe to import Tencent internals
137
  from hunyuanvideo_foley.utils.model_utils import load_model, denoise_process
138
  from hunyuanvideo_foley.utils.feature_utils import feature_process
139
  from hunyuanvideo_foley.utils.media_utils import merge_audio_video
 
163
  return d
164
 
165
  def auto_load_models() -> str:
166
+ """Load model natively (weights already downloaded to MODEL_PATH)."""
167
  global _model_dict, _cfg, _device
168
 
169
  if not os.path.exists(MODEL_PATH):