translation evaluation pipeline

2025-12-25 04:29:22 +01:00 · 2023-05-11 15:11:22 +08:00
parent 1c5222d6c1
commit d76303d324
1 changed files with 5 additions and 1 deletions
--- a/modelscope/pipelines/nlp/translation_evaluation_pipeline.py
+++ b/modelscope/pipelines/nlp/translation_evaluation_pipeline.py
@@ -42,7 +42,11 @@ class TranslationEvaluationPipeline(Pipeline):
                `"EvaluationMode.SRC"`, `"EvaluationMode.REF"`. Aside from hypothesis, the
                source/reference/source+reference can be presented during evaluation.
        """
-        super().__init__(model=model, preprocessor=preprocessor, **kwargs)
+        super().__init__(
+            model=model,
+            preprocessor=preprocessor,
+            compile=kwargs.pop('compile', False),
+            compile_options=kwargs.pop('compile_options', {}))

        self.eval_mode = eval_mode
        self.checking_eval_mode()