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

Simpler fix.

Browse Source
This commit is contained in:
Nicolas Patry 2023-09-26 13:03:45 +00:00
parent 649d9754b1
commit 9c0f679d1d
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/text_generation_server/cli.py
View File

@ -186,11 +186,7 @@ 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 = [] discard_names = getattr(class_, "_tied_weights_keys", [])
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 = []