From 78b8f9a66e8196fda234faa657d899bedb1ff747 Mon Sep 17 00:00:00 2001 From: "Wang, Chang" Date: Tue, 3 Dec 2024 18:29:03 +0800 Subject: [PATCH] Update utility.py --- neural_compressor/torch/utils/utility.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/neural_compressor/torch/utils/utility.py b/neural_compressor/torch/utils/utility.py index 5fb93a711fe..2ee4d9c9391 100644 --- a/neural_compressor/torch/utils/utility.py +++ b/neural_compressor/torch/utils/utility.py @@ -351,9 +351,8 @@ def load_empty_model(pretrained_model_name_or_path, cls=None, **kwargs): else: path = dowload_hf_model(pretrained_model_name_or_path) if cls.__base__ == _BaseAutoModelClass: - config = AutoConfig.from_pretrained(path, **kwargs) with init_empty_weights(): - model = cls.from_config(config, **kwargs) + model = cls.from_pretrained(path, **kwargs) else: # pragma: no cover config = cls.config_class.from_pretrained(path, **kwargs) with init_empty_weights():