Update inference.py

inference.py

@@ -29,6 +29,8 @@ class InferenceRecipe:
         # Initialize all model components
         logger.info(f"Initializing models from {model_path}")
         self.mimi, self.text_tokenizer, self.lm_gen = self._initialize_models()
+        self.mimi = self.mimi.to(self.device)
+        self.lm_gen = self.lm_gen.to(self.device)
         logger.info("Model initialization complete")
         
     def _initialize_models(self):