update models.py
Browse files- 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 = ""
|