From 7c8ee5814a1a4ced386fc39cc13b3caf83a30930 Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Tue, 28 Mar 2023 19:40:06 +0530
Subject: [PATCH] Update peft_model.py

---
 src/peft/peft_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py
index 71b9ef2..f2274fe 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -324,8 +324,8 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
             self.add_adapter(adapter_name, peft_config)
 
         # load weights if any
-        if kwargs.get("subfolder", None) is not None:
-            path = os.path.join(model_id, kwargs["subfolder"])
+        path = os.path.join(model_id, kwargs["subfolder"]) if kwargs.get("subfolder", None) is not None else model_id
+
         if os.path.exists(os.path.join(path, WEIGHTS_NAME)):
             filename = os.path.join(path, WEIGHTS_NAME)
         else: