converter: Use class name Rwkv6Model

Signed-off-by: Molly Sophia <[email protected]>
ggerganov · Aug 25, 2024 · d5380c6 · d5380c6
1 parent e4d5638
commit d5380c6
Showing 1 changed file with 1 addition and 1 deletion.
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):