Unverified Commit fd0b94fd authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`from_pretrained`] Fix failing PEFT tests (#25733)

fix failing PEFT tests
parent 1b2381c4
...@@ -2406,8 +2406,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2406,8 +2406,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
_commit_hash=commit_hash, _commit_hash=commit_hash,
) )
if _adapter_model_path is not None and os.path.isfile(_adapter_model_path): if _adapter_model_path is not None and os.path.isfile(_adapter_model_path):
with open(_adapter_model_path, "r", encoding="utf-8"): with open(_adapter_model_path, "r", encoding="utf-8") as f:
_adapter_model_path = pretrained_model_name_or_path _adapter_model_path = pretrained_model_name_or_path
pretrained_model_name_or_path = json.load(f)["base_model_name_or_path"]
# change device_map into a map if we passed an int, a str or a torch.device # change device_map into a map if we passed an int, a str or a torch.device
if isinstance(device_map, torch.device): if isinstance(device_map, torch.device):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment