aryo100 commited on
Commit
5b733b9
·
1 Parent(s): d42c4d2

update models.py

Browse files
Files changed (1) hide show
  1. modules/models.py +2 -1
modules/models.py CHANGED
@@ -241,7 +241,8 @@ class VoiceConvertModel:
241
  return os.path.join(MODELS_DIR, "checkpoints", f"{basename}.index")
242
 
243
 
244
- MODELS_DIR = opts.models_dir or os.path.join(ROOT_DIR, "models")
 
245
  vc_model: Optional[VoiceConvertModel] = None
246
  embedder_model: Optional[HubertModel] = None
247
  loaded_embedder_model = ""
 
241
  return os.path.join(MODELS_DIR, "checkpoints", f"{basename}.index")
242
 
243
 
244
+ # MODELS_DIR = opts.models_dir or os.path.join(ROOT_DIR, "models")
245
+ MODELS_DIR = os.environ.get("MODELS_DIR", "/data/models")
246
  vc_model: Optional[VoiceConvertModel] = None
247
  embedder_model: Optional[HubertModel] = None
248
  loaded_embedder_model = ""