ccibeekeoc42 commited on
Commit
46a9d71
·
verified ·
1 Parent(s): d086f71

Updated the wave tokenizer file paths

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,8 +94,8 @@ from yarngpt.audiotokenizer import AudioTokenizerV2
94
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
95
 
96
  tokenizer_path="saheedniyi/YarnGPT2"
97
- wav_tokenizer_config_path="/content/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
98
- wav_tokenizer_model_path = "/content/wavtokenizer_large_speech_320_24k.ckpt"
99
 
100
  audio_tokenizer=AudioTokenizerV2(tokenizer_path,wav_tokenizer_model_path,wav_tokenizer_config_path)
101
  tts_model = AutoModelForCausalLM.from_pretrained(tokenizer_path,torch_dtype="auto").to(audio_tokenizer.device)
 
94
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
95
 
96
  tokenizer_path="saheedniyi/YarnGPT2"
97
+ wav_tokenizer_config_path = "./wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
98
+ wav_tokenizer_model_path = "./wavtokenizer_large_speech_320_24k.ckpt"
99
 
100
  audio_tokenizer=AudioTokenizerV2(tokenizer_path,wav_tokenizer_model_path,wav_tokenizer_config_path)
101
  tts_model = AutoModelForCausalLM.from_pretrained(tokenizer_path,torch_dtype="auto").to(audio_tokenizer.device)