Update loader.py

Former-commit-id: 3b229a27a108b840e6bed3c8684737f51ce9faf4
This commit is contained in:
hoshi-hiyouga 2024-10-17 19:48:12 +08:00 committed by GitHub
parent 66819fd3ee
commit c6e1313977

View File

@ -116,7 +116,6 @@ def load_config(model_args: "ModelArguments") -> "PretrainedConfig":
Loads model config. Loads model config.
""" """
init_kwargs = _get_init_kwargs(model_args) init_kwargs = _get_init_kwargs(model_args)
return AutoConfig.from_pretrained(model_args.model_name_or_path, **init_kwargs) return AutoConfig.from_pretrained(model_args.model_name_or_path, **init_kwargs)