vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add support for starcoder2

Browse source
This commit is contained in:
Cocoa 2024-03-02 00:20:49 +08:00
parent e743386728
commit e8e6103f42
No known key found for this signature in database
GPG key ID: D731045A69565F10
2 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert-hf-to-gguf.py
View file

@ -249,6 +249,8 @@ class Model:
return gguf.MODEL_ARCH.FALCON return gguf.MODEL_ARCH.FALCON
if arch == "GPTBigCodeForCausalLM": if arch == "GPTBigCodeForCausalLM":
return gguf.MODEL_ARCH.STARCODER return gguf.MODEL_ARCH.STARCODER
if arch == "Starcoder2ForCausalLM":
return gguf.MODEL_ARCH.LLAMA
if arch == "GPTRefactForCausalLM": if arch == "GPTRefactForCausalLM":
return gguf.MODEL_ARCH.REFACT return gguf.MODEL_ARCH.REFACT
if arch == "PersimmonForCausalLM": if arch == "PersimmonForCausalLM":

2
gguf-py/gguf/tensor_mapping.py
View file

@ -210,6 +210,7 @@ class TensorNameMap:
"model.layers.layers.{bid}.mlp.up_proj", # plamo "model.layers.layers.{bid}.mlp.up_proj", # plamo
"model.layers.{bid}.feed_forward.w3", # internlm2 "model.layers.{bid}.feed_forward.w3", # internlm2
"encoder.layers.{bid}.mlp.fc11", # nomic-bert "encoder.layers.{bid}.mlp.fc11", # nomic-bert
"model.layers.{bid}.mlp.c_fc", # starcoder2
), ),
MODEL_TENSOR.FFN_UP_EXP: ( MODEL_TENSOR.FFN_UP_EXP: (
@ -256,6 +257,7 @@ class TensorNameMap:
"model.layers.layers.{bid}.mlp.down_proj", # plamo "model.layers.layers.{bid}.mlp.down_proj", # plamo
"model.layers.{bid}.feed_forward.w2", # internlm2 "model.layers.{bid}.feed_forward.w2", # internlm2
"encoder.layers.{bid}.mlp.fc2", # nomic-bert "encoder.layers.{bid}.mlp.fc2", # nomic-bert
"model.layers.{bid}.mlp.c_proj", # starcoder2
), ),
MODEL_TENSOR.FFN_DOWN_EXP: ( MODEL_TENSOR.FFN_DOWN_EXP: (