Changed embed_model_path to emb_model_path

DigitalPhonetics · Dec 24, 2023 · 8a20e01 · 8a20e01
1 parent ddb1d80
commit 8a20e01
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/anonymization/modules/speaker_embeddings/anonymization/pool_anon.py b/anonymization/modules/speaker_embeddings/anonymization/pool_anon.py
@@ -149,15 +149,15 @@ def __init__(
         self.scaling = scaling
         self.stats_per_dim_path = stats_per_dim_path or Path()
 
-    def _load_pool_embeddings(self, pool_data_dir, pool_vec_path, embed_model_path):
+    def _load_pool_embeddings(self, pool_data_dir, pool_vec_path, emb_model_path):
         logger.debug(pool_data_dir)
         if pool_vec_path.exists():
             pool_embeddings = SpeakerEmbeddings(
                 vec_type=self.vec_type, emb_level="spk", device=self.device
             )
             pool_embeddings.load_vectors(pool_vec_path)
         else:
-            extraction_settings = {"vec_type": self.vec_type, "emb_level": "spk", "embed_model_path": embed_model_path}
+            extraction_settings = {"vec_type": self.vec_type, "emb_level": "spk", "emb_model_path": emb_model_path}
             emb_extractor = SpeakerExtraction(
                 results_dir=pool_vec_path,
                 devices=[self.device],

diff --git a/configs/anon/pool.yaml b/configs/anon/pool.yaml
@@ -1,6 +1,6 @@
 !new:anonymization.modules.speaker_embeddings.anonymization.pool_anon.PoolAnonymizer
   data_dir: !PLACEHOLDER # to be overridden by the main config
-  embed_model_path: !PLACEHOLDER # to be overridden by the main config
+  emb_model_path: !PLACEHOLDER # to be overridden by the main config
   vectors_dir: !PLACEHOLDER # to be overridden by the main config
   pool_data_dir: !ref <data_dir>/libritts_train_other_500
   pool_vec_path: !ref <vectors_dir>/style-embed_spk-level/pool_embeddings