Update voila_tokenizer.py

voila_tokenizer.py

@@ -24,7 +24,11 @@ class VoilaTokenizer:
         self.sampling_rate = self.processor.sampling_rate
         self.model_version = self.model.config.model_version
 
-
+    def to(self, device):
+        self.device = torch.device(device)
+        self.model = self.model.to(device)
+        self.bandwidth_id = self.bandwidth_id.to(device)
+        
     @torch.no_grad()
     def encode(self, wav, sr):
         wav = torch.tensor(wav, dtype=torch.float32, device=self.device)