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

converter: Use class name `Rwkv6Model`

Browse source 
Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
This commit is contained in:
Molly Sophia 2024-08-25 15:56:43 +08:00
parent 7756afd8dd
commit 87a29014a4
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert_hf_to_gguf.py
View file

@ -2719,7 +2719,7 @@ class StarCoder2Model(Model):
@Model.register("Rwkv6ForCausalLM") @Model.register("Rwkv6ForCausalLM")
class RwkvModel(Model): class Rwkv6Model(Model):
model_arch = gguf.MODEL_ARCH.RWKV6 model_arch = gguf.MODEL_ARCH.RWKV6
def set_vocab(self): def set_vocab(self):