Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix logger info

Browse Source
This commit is contained in:
suluyan.sly
2023-12-05 15:21:32 +08:00
parent 954bd64919
commit b963dd3b32
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/pipelines/nlp/llm_pipeline.py
View File

@@ -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,