From d5380c6ee2aeb1b10505b1ebc831d4596d590450 Mon Sep 17 00:00:00 2001
From: Molly Sophia <mollysophia379@gmail.com>
Date: Sun, 25 Aug 2024 15:56:43 +0800
Subject: [PATCH] converter: Use class name ``Rwkv6Model``

Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
---
 convert_hf_to_gguf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
index dd0f5c7dd03da..541a430eed8d0 100755
--- 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):