Update utils/interface.py

utils/interface.py

@@ -33,8 +33,8 @@ class Inference:
             trust_remote_code=True,
         )
         self.model = AutoModel.from_pretrained(
-            model_path, trust_remote_code=True, device_map="auto"
-        ).eval()
+            model_path, trust_remote_code=True
+        ).to(self.device).eval()
 
     def forward(
         self,
@@ -196,7 +196,7 @@ class MIMOInterface:
         try:
             # --- Handle Decoder Audio Prompt ---
             decoder_audio_prompt_path_for_model = None
-            if decoder_audio_prompt is not None:
+            if decoder_audio_prompt:
                 saved_decoder_audio_path = self._save_audio_numpy(decoder_audio_prompt, prefix="decoder_prompt")
                 audio_file_paths_to_delete.append(saved_decoder_audio_path)
                 decoder_audio_prompt_path_for_model = saved_decoder_audio_path