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

tied embeddings for qwe2

Browse Source
This commit is contained in:
matvey-kolbasov-hs 2024-07-24 11:59:56 +03:00
parent 8642250602
commit f73f57ca21
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/text_generation_server/models/__init__.py
View File

@ -959,6 +959,10 @@ def get_model(
dtype=dtype,
trust_remote_code=trust_remote_code,
lora_adapter_ids=lora_adapter_ids,
aliases={
"lm_head.weight": ["model.word_embeddings.weight"],
"model.word_embeddings.weight": ["lm_head.weight"],
}
)
elif sharded:
raise NotImplementedError(FLASH_ATT_ERROR_MESSAGE.format("Sharded Qwen2"))