Fix paths

Darwinkel · Feb 26, 2024 · 81246f0 · 81246f0
1 parent 1f1b28b
commit 81246f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/code/settings.py b/code/settings.py
@@ -3,9 +3,9 @@
 
 import torch
 
+CHECKPOINT_DIR = "/xtts_model"
 CHECKPOINTS_CONFIG_JSON = "/xtts_model/config.json"
-CHECKPOINT_DIR = "/xtts_model/current_model"
-CHECKPOINT_VOCAB = "/xtts_model/current_model/vocab.json"
+CHECKPOINT_VOCAB_JSON = "/xtts_model/vocab.json"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 GRADIO_EXAMPLES_CACHE = os.environ["GRADIO_EXAMPLES_CACHE"]

diff --git a/code/utils.py b/code/utils.py
@@ -21,7 +21,7 @@ def load_checkpoint():
     model.load_checkpoint(
         config,
         checkpoint_dir=settings.CHECKPOINT_DIR,
-        vocab_path=settings.CHECKPOINT_VOCAB,
+        vocab_path=settings.CHECKPOINT_VOCAB_JSON,
         use_deepspeed=use_deepspeed,
     )
     model.to(settings.DEVICE)