VITA-Audio

Runtime error

shenyunhang commited on May 6

Commit

9e0adab

verified ·

1 Parent(s): 46ab98e

Update vita_audio/tokenizer_sensevoice_glm4voice.py

Files changed (1) hide show

vita_audio/tokenizer_sensevoice_glm4voice.py CHANGED Viewed

@@ -115,7 +115,7 @@ class SenseVoiceGLM4VoiceTokenizer:
         if self.rank is not None:
             self.device = f"cuda:{self.rank}"
-            torch.cuda.set_device(self.rank)
         else:
             self.device = "cpu"

         if self.rank is not None:
             self.device = f"cuda:{self.rank}"
+            #torch.cuda.set_device(self.rank)
         else:
             self.device = "cpu"