mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-11 04:14:52 +00:00
Fix local load for peft
local directory overloaded still needs the directory to locate the weights files correctly.
This commit is contained in:
parent
564199bab3
commit
83f81a6b89
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user