ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

refine (#8912)

Browse source 
Co-authored-by: leonardozcm <leonardozcm@gmail.com>
This commit is contained in:
Zhao Changmin 2023-09-11 16:40:33 +08:00 committed by GitHub
parent df165ad165
commit e62eda74b8
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/llm/src/bigdl/llm/transformers/convert.py
View file

@ -134,8 +134,8 @@ def optimize(model):
# All huggingface format models are inherited from `PreTrainedModel` # All huggingface format models are inherited from `PreTrainedModel`
if not isinstance(model, PreTrainedModel): if not isinstance(model, PreTrainedModel):
logger.info("It is not supported that optimizing a model isn't belong to" logger.info("Only HuggingFace Transformers models are currently "
"huggingface transformers with `optimize_model=True` for now.") "supported for further optimizations")
return model return model
trans_version = transformers.__version__ trans_version = transformers.__version__