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

fix discard_names in safetensors convertion

Browse Source
This commit is contained in:
zhangsibo1129 2023-09-25 10:43:28 +08:00
parent 123749a3c9
commit 649d9754b1
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/text_generation_server/cli.py
View File

@ -186,8 +186,11 @@ def download_weights(
class_ = getattr(transformers, architecture) class_ = getattr(transformers, architecture)
# Name for this varible depends on transformers version. # Name for this varible depends on transformers version.
discard_names = getattr(class_, "_tied_weights_keys", []) discard_names = []
discard_names.extend(getattr(class_, "_keys_to_ignore_on_load_missing", [])) if getattr(class_, "_tied_weights_keys", []):
discard_names.extend(getattr(class_, "_tied_weights_keys", []))
if getattr(class_, "_keys_to_ignore_on_load_missing", []):
discard_names.extend(getattr(class_, "_keys_to_ignore_on_load_missing", []))
except Exception as e: except Exception as e:
discard_names = [] discard_names = []