okewunmi commited on
Commit
1816e2d
·
verified ·
1 Parent(s): 06a1ff8

Update yarngpt_utils.py

Browse files
Files changed (1) hide show
  1. yarngpt_utils.py +3 -3
yarngpt_utils.py CHANGED
@@ -10,10 +10,10 @@ class AudioTokenizer:
10
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
11
  self.tokenizer = AutoTokenizer.from_pretrained(hf_path)
12
 
13
- # Correct initialization for WavTokenizer
14
  self.wav_tokenizer = WavTokenizer(
15
- model_path=wav_tokenizer_model_path,
16
- config_path=wav_tokenizer_config_path,
17
  device=self.device
18
  )
19
 
 
10
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
11
  self.tokenizer = AutoTokenizer.from_pretrained(hf_path)
12
 
13
+ # The correct initialization for WavTokenizer
14
  self.wav_tokenizer = WavTokenizer(
15
+ config=wav_tokenizer_config_path,
16
+ model=wav_tokenizer_model_path,
17
  device=self.device
18
  )
19