Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit 0fc6e01

Browse files
authored
Update modeling_auto.py (#1499)
Signed-off-by: Wang, Chang <[email protected]>
1 parent 307ab8e commit 0fc6e01

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intel_extension_for_transformers/transformers/modeling/modeling_auto.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1032,7 +1032,7 @@ def load_low_bit(cls, pretrained_model_name_or_path, *model_args, **kwargs):
10321032
variant = kwargs.pop("variant", None)
10331033
offload_folder = kwargs.pop("offload_folder", None)
10341034
offload_state_dict = kwargs.pop("offload_state_dict", False)
1035-
torch_dtype = kwargs.get("torch_dtype", "auto")
1035+
torch_dtype = kwargs.pop("torch_dtype", "auto")
10361036
cache_dir = kwargs.pop("cache_dir", None)
10371037
force_download = kwargs.pop("force_download", False)
10381038
proxies = kwargs.pop("proxies", None)

0 commit comments

Comments
 (0)