From d76303d324e747196587578e5ea24d81a6798ae3 Mon Sep 17 00:00:00 2001 From: slin000111 Date: Thu, 11 May 2023 15:11:22 +0800 Subject: [PATCH] translation evaluation pipeline --- modelscope/pipelines/nlp/translation_evaluation_pipeline.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/nlp/translation_evaluation_pipeline.py b/modelscope/pipelines/nlp/translation_evaluation_pipeline.py index a9e929d1..8a339517 100644 --- 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()