From bbef56c17ac64155cb22d70c164be95e350410ac Mon Sep 17 00:00:00 2001 From: Kaihui-intel Date: Fri, 11 Jul 2025 13:10:45 +0800 Subject: [PATCH] add is_remote_code for auto_class Signed-off-by: Kaihui-intel --- neural_compressor/transformers/models/modeling_auto.py | 1 + 1 file changed, 1 insertion(+) diff --git a/neural_compressor/transformers/models/modeling_auto.py b/neural_compressor/transformers/models/modeling_auto.py index 364b991caa8..9a8e5d62a72 100644 --- a/neural_compressor/transformers/models/modeling_auto.py +++ b/neural_compressor/transformers/models/modeling_auto.py @@ -409,6 +409,7 @@ def load_low_bit(cls, pretrained_model_name_or_path, *model_args, **kwargs): user_agent=user_agent, revision=revision, commit_hash=commit_hash, + is_remote_code=True, ) is_sharded = sharded_metadata is not None resolved_archive_file = checkpoint_files if is_sharded else checkpoint_files[0]