All model to have device changed

cli/SparkTTS.py

@@ -49,6 +49,11 @@ class SparkTTS:
         self.model = AutoModelForCausalLM.from_pretrained(f"{self.model_dir}/LLM")
         self.audio_tokenizer = BiCodecTokenizer(self.model_dir, device=self.device)
         self.model.to(self.device)
+        
+    def to(self, device: torch.device):
+        self.device = device
+        self.model.to(self.device)
+        self.audio_tokenizer.to(self.device)
 
     def process_prompt(
         self,