Update yarngpt_utils.py

yarngpt_utils.py

@@ -10,10 +10,10 @@ class AudioTokenizer:
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.tokenizer = AutoTokenizer.from_pretrained(hf_path)
         
-        # The correct initialization for WavTokenizer
+        # Fix: Use the correct parameter names for WavTokenizer
         self.wav_tokenizer = WavTokenizer(
-            config=wav_tokenizer_config_path,
-            model=wav_tokenizer_model_path,
+            checkpoint_path=wav_tokenizer_model_path,
+            config_path=wav_tokenizer_config_path,
             device=self.device
         )