Update model.py

Samagra-Development · Apr 9, 2024 · 85c736b · 85c736b
1 parent 1f4d19d
commit 85c736b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/asr/fairseq_mms/local/model.py b/src/asr/fairseq_mms/local/model.py
@@ -26,6 +26,7 @@ async def inference(self,  request: ModelRequest):
         wav_file = request.wav_file
         ory_sample, sr = librosa.load(wav_file, sr=16000)
         inputs = self.processor(ory_sample, sampling_rate=16_000, return_tensors="pt")
+        inputs = inputs.to(self.device)
         with torch.no_grad():
             outputs = self.model(**inputs).logits