From 83f81a6b89de038091ce8465bfbbf7259b0a111e Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 21 Dec 2023 15:35:15 +0000
Subject: [PATCH] Fix local load for peft

local directory overloaded still needs the directory to locate the
weights files correctly.
---
 server/text_generation_server/utils/hub.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/text_generation_server/utils/hub.py b/server/text_generation_server/utils/hub.py
index deb1a941..b56484f6 100644
--- a/server/text_generation_server/utils/hub.py
+++ b/server/text_generation_server/utils/hub.py
@@ -49,7 +49,7 @@ def _weight_files_from_dir(d: Path, extension: str) -> List[str]:
     # done there with the len(s.rfilename.split("/")) == 1 condition
     root, _, files = next(os.walk(str(d)))
     filenames = [
-        f
+        os.path.join(root, f)
         for f in files
         if f.endswith(extension)
         and "arguments" not in f