nac model test

anonymization/modules/nac/anonymizer.py

@@ -19,7 +19,9 @@ class Anonymizer(torch.nn.Module):
         super().__init__()
 
         if checkpoint_dir is None:
-            checkpoint_dir = os.path.expanduser('~/.local/share/tts/tts_models--multilingual--multi-dataset--bark')
+            checkpoint_dir = 'exp/nac_models'
+        if not os.path.exists(checkpoint_dir):
+            os.makedirs(checkpoint_dir)
 
         # 1. initialize Bark
         config = BarkConfig()  # don't change the custom config for the love of god
@@ -77,4 +79,4 @@ class Anonymizer(torch.nn.Module):
         audio_arr, _, _ = self.model.semantic_to_waveform(
             semantic_tokens, history_prompt=history_prompt, temp=coarse_temperature
         )
-        return audio_arr
\ No newline at end of file
+        return audio_arr

anonymization/pipelines/nac/install.sh

@@ -16,7 +16,7 @@ if [ ! -d exp/nac_mappings ]; then
     echo "Download pre-computed speaker mappings for the NAC pipeline"
     unzip_location="exp/nac_mappings"
     zip_file="./anonymization/modules/nac/speaker_mappings.zip"
-    curl -q -L -o $zip_file https://github.com/Voice-Privacy-Challenge/Voice-Privacy-Challenge-2024/releases/download/nac_mappings.zip/nac_speaker_mappings.zip
+    curl -q -L -o $zip_file https://github.com/m-pana/nac-requirements/releases/download/1.0/nac_speaker_mappings.zip
     unzip $zip_file -d $unzip_location
     rm $zip_file
 fi

configs/anon_nac.yaml

@@ -21,7 +21,7 @@ datasets:
 
 modules:
   model:
-    checkpoint_dir: ~/.local/share/tts/tts_models--multilingual--multi-dataset--bark
+    checkpoint_dir: exp/nac_models
     voice_dir: exp/nac_mappings/suno_voices/v2
 
   speaker_mappings_root: exp/nac_mappings/speaker_mappings