saheedniyi
/

YarnGPT2

text-generation

text-generation-inference

Model card Files Files and versions

saheedniyi commited on Feb 27

Commit

343d40a

·

verified ·

1 Parent(s): c99a1c2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ import torchaudio
 import IPython
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from outetts.wav_tokenizer.decoder import WavTokenizer
-from yarngpt.audiotokenizer import AudioTokenizer
 # download the wavtokenizer weights and config (to encode and decode the audio)
@@ -76,7 +76,7 @@ wav_tokenizer_config_path="/content/wavtokenizer_mediumdata_frame75_3s_nq1_code4
 wav_tokenizer_model_path = "/content/wavtokenizer_large_speech_320_24k.ckpt"
 # create the AudioTokenizer object
-audio_tokenizer=AudioTokenizer(
     hf_path,wav_tokenizer_model_path,wav_tokenizer_config_path
 )

 import IPython
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from outetts.wav_tokenizer.decoder import WavTokenizer
+from yarngpt.audiotokenizer import AudioTokenizerV2
 # download the wavtokenizer weights and config (to encode and decode the audio)
 wav_tokenizer_model_path = "/content/wavtokenizer_large_speech_320_24k.ckpt"
 # create the AudioTokenizer object
+audio_tokenizer=AudioTokenizerV2(
     hf_path,wav_tokenizer_model_path,wav_tokenizer_config_path
 )