Spaces:

shenyunhang
/

VITA-Audio

Running on Zero

shenyunhang commited on May 6

Commit

ff6efdc

verified ·

1 Parent(s): af4ebf7

Update vita_audio/tokenizer_sensevoice_glm4voice.py

Files changed (1) hide show

vita_audio/tokenizer_sensevoice_glm4voice.py CHANGED Viewed

@@ -120,7 +120,7 @@ class SenseVoiceGLM4VoiceTokenizer:
             self.device = "cpu"
         logger.info(f"{self.device=} Loading SenseVoiceSmall")
-        model_dir = "/data/models/FunAudioLLM/SenseVoiceSmall/"
         _, self.kwargs = SenseVoiceSmall.from_pretrained(model=model_dir, device=self.device)
         logger.info(f"{self.device=} Loading SenseVoiceSmall Done")

             self.device = "cpu"
         logger.info(f"{self.device=} Loading SenseVoiceSmall")
+        model_dir = "FunAudioLLM/SenseVoiceSmall/"
         _, self.kwargs = SenseVoiceSmall.from_pretrained(model=model_dir, device=self.device)
         logger.info(f"{self.device=} Loading SenseVoiceSmall Done")