huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 04:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt

Browse Source
This commit is contained in:
OlivierDehaene 2023-12-21 11:35:06 +01:00
parent 38df4c1d67
commit 672f290901
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/text_generation_server/utils/hub.py
View File

@ -48,12 +48,15 @@ def _weight_files_from_dir(d: Path, extension: str) -> List[str]:
# see _weight_hub_files_from_model_info, that's also what is # see _weight_hub_files_from_model_info, that's also what is
# done there with the len(s.rfilename.split("/")) == 1 condition # done there with the len(s.rfilename.split("/")) == 1 condition
root, _, files = next(os.walk(str(d))) root, _, files = next(os.walk(str(d)))
filenames = [f for f in files filenames = [
f
for f in files
if f.endswith(extension) if f.endswith(extension)
and "arguments" not in f and "arguments" not in f
and "args" not in f and "args" not in f
and "adapter" not in f and "adapter" not in f
and "training" not in f] and "training" not in f
]
return filenames return filenames