Spaces:
Sleeping
Sleeping
Updated the wave tokenizer file paths
Browse files
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="
|
98 |
-
wav_tokenizer_model_path = "
|
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)
|