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

Adding new ignore_rule for conversion.

Browse Source 
Hesitant to whitelist `pytorch_model` instead tbh.
This commit is contained in:
Nicolas Patry 2023-06-22 21:37:10 +02:00
parent c9c65ab323
commit 09e4e73f6e
No known key found for this signature in database
GPG Key ID: 6AE76ACC68FFBAF9
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -30,6 +30,7 @@ def weight_hub_files(
and len(s.rfilename.split("/")) == 1 and len(s.rfilename.split("/")) == 1
and "arguments" not in s.rfilename and "arguments" not in s.rfilename
and "args" not in s.rfilename and "args" not in s.rfilename
and "training" not in s.rfilename
] ]
if not filenames: if not filenames: