diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 44213f2f3..b1e7992f8 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -2719,7 +2719,7 @@ class StarCoder2Model(Model): @Model.register("Rwkv6ForCausalLM") -class RwkvModel(Model): +class Rwkv6Model(Model): model_arch = gguf.MODEL_ARCH.RWKV6 def set_vocab(self):