From 915a5db0c60b2d9fc7a64d37ced24bf0a348408b Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Wed, 1 Feb 2023 16:25:42 +0530 Subject: [PATCH] fixes --- src/peft/peft_model.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py index eba63a6..af17075 100644 --- a/src/peft/peft_model.py +++ b/src/peft/peft_model.py @@ -96,7 +96,11 @@ class PeftModel(PushToHubMixin, torch.nn.Module): # save the config if self.peft_config.base_model_name_or_path is None: - self.peft_config.base_model_name_or_path = self.base_model.__dict__.get("name_or_path", None) + self.peft_config.base_model_name_or_path = ( + self.base_model.__dict__.get("name_or_path", None) + if isinstance(self.peft_config, PromptLearningConfig) + else self.base_model.model.__dict__.get("name_or_path", None) + ) self.peft_config.inference_mode = True self.peft_config.save_pretrained(save_directory)