Update srt_model.py

srt_model.py

@@ -268,7 +268,7 @@ class CustomSLM(PreTrainedModel):
 
 
     @torch.no_grad()
-    def translate_encode(self, beam_search, mels, prompts, max_new_tokens=300, use_vllm=False):
+    def translate_encode(self, mels):
         device = self.device
         dtype = self.dtype