Spaces:
Running
on
Zero
Running
on
Zero
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 |
-
|
|
|
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 |
|