From e62eda74b830ff580a6c7151352d1db32c1a9d77 Mon Sep 17 00:00:00 2001 From: Zhao Changmin Date: Mon, 11 Sep 2023 16:40:33 +0800 Subject: [PATCH] refine (#8912) Co-authored-by: leonardozcm --- python/llm/src/bigdl/llm/transformers/convert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/llm/src/bigdl/llm/transformers/convert.py b/python/llm/src/bigdl/llm/transformers/convert.py index 6fd3d406..00156300 100644 --- a/python/llm/src/bigdl/llm/transformers/convert.py +++ b/python/llm/src/bigdl/llm/transformers/convert.py @@ -134,8 +134,8 @@ def optimize(model): # All huggingface format models are inherited from `PreTrainedModel` if not isinstance(model, PreTrainedModel): - logger.info("It is not supported that optimizing a model isn't belong to" - "huggingface transformers with `optimize_model=True` for now.") + logger.info("Only HuggingFace Transformers models are currently " + "supported for further optimizations") return model trans_version = transformers.__version__