shenyunhang commited on
Commit
2e51c14
·
verified ·
1 Parent(s): c443173

Update vita_audio/tokenizer_sensevoice_glm4voice.py

Browse files
vita_audio/tokenizer_sensevoice_glm4voice.py CHANGED
@@ -120,7 +120,8 @@ class SenseVoiceGLM4VoiceTokenizer:
120
  self.device = "cpu"
121
 
122
  logger.info(f"{self.device=} Loading SenseVoiceSmall")
123
- model_dir = "FunAudioLLM/SenseVoiceSmall"
 
124
  _, self.kwargs = SenseVoiceSmall.from_pretrained(model=model_dir, device=self.device)
125
  logger.info(f"{self.device=} Loading SenseVoiceSmall Done")
126
 
 
120
  self.device = "cpu"
121
 
122
  logger.info(f"{self.device=} Loading SenseVoiceSmall")
123
+ from huggingface_hub import snapshot_download
124
+ model_dir = snapshot_download(repo_id="FunAudioLLM/SenseVoiceSmall")
125
  _, self.kwargs = SenseVoiceSmall.from_pretrained(model=model_dir, device=self.device)
126
  logger.info(f"{self.device=} Loading SenseVoiceSmall Done")
127