converter: Use class name Rwkv6Model

MollySophia · MollySophia · commit 7c6e52033020 · 2024-08-25T17:36:31.000+08:00
Signed-off-by: Molly Sophia &lt;mollysophia379@gmail.com&gt;
diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
@@ -2715,7 +2715,7 @@ class StarCoder2Model(Model):
 
 
 @Model.register("Rwkv6ForCausalLM")
-class RwkvModel(Model):
+class Rwkv6Model(Model):
     model_arch = gguf.MODEL_ARCH.RWKV6
 
     def set_vocab(self):
