diff --git a/python/llm/src/bigdl/llm/serving/bigdl_llm_model.py b/python/llm/src/bigdl/llm/serving/bigdl_llm_model.py index c4716b5e..fafb0a36 100644 --- a/python/llm/src/bigdl/llm/serving/bigdl_llm_model.py +++ b/python/llm/src/bigdl/llm/serving/bigdl_llm_model.py @@ -243,7 +243,7 @@ class BigDLLLMAdapter(BaseModelAdapter): def load_model(self, model_path: str, from_pretrained_kwargs: dict): revision = from_pretrained_kwargs.get("revision", "main") tokenizer = AutoTokenizer.from_pretrained( - model_path, use_fast=False, revision=revision + model_path, use_fast=False, revision=revision, trust_remote_code=True ) print("Customized bigdl-llm loader") from bigdl.llm.transformers import AutoModelForCausalLM