Bils commited on
Commit
f8f20d3
·
verified ·
1 Parent(s): 2865461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import os, sys, json, tempfile, subprocess, shutil, uuid
2
  from pathlib import Path
3
  from typing import Optional, Tuple, List
4
 
@@ -113,9 +113,31 @@ def prepare_once():
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 "
@@ -176,6 +198,10 @@ def auto_load_models() -> str:
176
  logger.info("Loading HunyuanVideo-Foley model...")
177
  logger.info(f"MODEL_PATH: {MODEL_PATH}")
178
  logger.info(f"CONFIG_PATH: {CONFIG_PATH}")
 
 
 
 
179
  _model_dict, _cfg = load_model(MODEL_PATH, CONFIG_PATH, _device)
180
  logger.info("✅ Model loaded")
181
  return "✅ Model loaded"
 
1
+ import os, sys, json, tempfile, subprocess, shutil, uuid, glob
2
  from pathlib import Path
3
  from typing import Optional, Tuple, List
4
 
 
113
 
114
  prepare_once()
115
 
116
+ # Prefer safetensors globally (avoids bad *.bin)
117
+ os.environ["TRANSFORMERS_PREFER_SAFETENSORS"] = "1"
118
 
119
+ def _purge_clap_pt_bins():
120
+ """
121
+ Remove any cached pytorch_model.bin for laion/larger_clap_general so
122
+ Transformers loads the safetensors instead.
123
+ """
124
+ cache_root = Path.home() / ".cache" / "huggingface" / "hub"
125
+ patterns = [
126
+ cache_root / "models--laion--larger_clap_general" / "snapshots" / "*" / "pytorch_model.bin",
127
+ cache_root / "models--laion--larger_clap_general" / "snapshots" / "*" / "model.bin",
128
+ cache_root / "models--laion--larger_clap_general" / "snapshots" / "*" / "*.bin",
129
+ ]
130
+ for pat in patterns:
131
+ for f in glob.glob(str(pat)):
132
+ try:
133
+ Path(f).unlink()
134
+ print(f">> Purged cached bin: {f}")
135
+ except Exception:
136
+ pass
137
+
138
+ # ---- Friendly dependency checks (clear errors during boot) -------------------
139
  try:
140
+ import audiotools # provided by the PyPI package 'descript-audiotools'
141
  except Exception as e:
142
  raise RuntimeError(
143
  "Missing module 'audiotools'. Install it via the PyPI package "
 
198
  logger.info("Loading HunyuanVideo-Foley model...")
199
  logger.info(f"MODEL_PATH: {MODEL_PATH}")
200
  logger.info(f"CONFIG_PATH: {CONFIG_PATH}")
201
+
202
+ # Make sure any corrupted *.bin in CLAP cache is removed
203
+ _purge_clap_pt_bins()
204
+
205
  _model_dict, _cfg = load_model(MODEL_PATH, CONFIG_PATH, _device)
206
  logger.info("✅ Model loaded")
207
  return "✅ Model loaded"