Update chatterbox/tts_turbo.py

chatterbox/tts_turbo.py

@@ -129,6 +129,15 @@ class ChatterboxTurboTTS:
         self.conds = conds
         self.watermarker = perth.PerthImplicitWatermarker()
 
+    def to(self, device):
+        self.device = device
+        self.t3 = self.t3.to(device)
+        self.s3gen = self.s3gen.to(device)
+        self.ve = self.ve.to(device)
+        if self.conds is not None:
+            self.conds = self.conds.to(device)
+        return self
+    
     @classmethod
     def from_local(cls, ckpt_dir, device) -> 'ChatterboxTurboTTS':
         ckpt_dir = Path(ckpt_dir)
@@ -244,15 +253,6 @@ class ChatterboxTurboTTS:
             emotion_adv=exaggeration * torch.ones(1, 1, 1),
         ).to(device=self.device)
         self.conds = Conditionals(t3_cond, s3gen_ref_dict)
-
-    def to(self, device):
-        self.device = device
-        self.t3 = self.t3.to(device)
-        self.s3gen = self.s3gen.to(device)
-        self.ve = self.ve.to(device)
-        if self.conds is not None:
-            self.conds = self.conds.to(device)
-        return self
     
     def generate(
         self,