From 6dc374782bf675573f8c8ebee9312e9c55567bf2 Mon Sep 17 00:00:00 2001 From: Mehdi Lahlou Date: Fri, 19 Apr 2024 18:45:44 +0200 Subject: [PATCH] Fixed server not passing sequence `ref`/`tgt` to `translator.translate` method --- onmt/translate/translation_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onmt/translate/translation_server.py b/onmt/translate/translation_server.py index 6dbb269153..67b52f59f0 100644 --- a/onmt/translate/translation_server.py +++ b/onmt/translate/translation_server.py @@ -596,7 +596,7 @@ def run(self, inputs): if len(examples) > 0: try: if isinstance(self.translator, CTranslate2Translator): - scores, predictions = self.translator.translate(examples) + scores, predictions = self.translator.translate(examples, tgt=[item["tgt"]["tgt"] if item["tgt"] is not None else "" for item in examples]) else: device_id = ( self.translator._dev.index if self.translator._use_cuda else -1