From b963dd3b32cacb9c30696e59bf224c6d7f786000 Mon Sep 17 00:00:00 2001 From: "suluyan.sly" Date: Tue, 5 Dec 2023 15:21:32 +0800 Subject: [PATCH] fix logger info --- modelscope/pipelines/nlp/llm_pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/pipelines/nlp/llm_pipeline.py b/modelscope/pipelines/nlp/llm_pipeline.py index a0791c31..e9606560 100644 --- a/modelscope/pipelines/nlp/llm_pipeline.py +++ b/modelscope/pipelines/nlp/llm_pipeline.py @@ -110,13 +110,13 @@ class LLMPipeline(Pipeline): try: model = self._wrap_infer_framework(model_dir, self.llm_framework) - logger.info(f'initiate model with {framework}.') + logger.info(f'initiate model with {self.llm_framework}.') return model except Exception as e: - self.llm_framework = None logger.warning( f'Cannot using llm_framework with {model}, ' f'ignoring llm_framework={self.llm_framework} : {e}') + self.llm_framework = None if is_model(model): return Model.from_pretrained( model,