Upload folder using huggingface_hub

config.json

@@ -2,6 +2,10 @@
   "architectures": [
     "Rwkv5ForCausalLM"
   ],
+  "auto_map": {
+      "AutoConfig": "configuration_rwkv5.Rwkv5Config",
+      "AutoModelForCausalLM": "modeling_rwkv5.Rwkv5ForCausalLM"
+  },
   "attention_hidden_size": 2560,
   "bos_token_id": 0,
   "eos_token_id": 0,

modeling_rwkv5.py

@@ -436,7 +436,7 @@ class Rwkv5PreTrainedModel(PreTrainedModel):
     """
 
     config_class = Rwkv5Config
-    base_model_prefix = "rwkv"
+    base_model_prefix = "rwkv5"
     _no_split_modules = ["RwkvBlock"]
     _keep_in_fp32_modules = ["time_decay", "time_first"]
     supports_gradient_checkpointing = True