add model

7d286ad0 · Baber · 24214440 · 7d286ad0 · 7d286ad0
Commit 7d286ad0 authored Jan 23, 2025 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

lm_eval/models/__init__.py lm_eval/models/__init__.py +1 -0

lm_eval/models/rwkvwrapper.py lm_eval/models/rwkvwrapper.py +1 -1

No files found.
--- a/lm_eval/models/__init__.py
+++ b/lm_eval/models/__init__.py
@@ -13,6 +13,7 @@ from . import (
    openai_completions,
    optimum_ipex,
    optimum_lm,
+    rwkvwrapper,
    textsynth,
    vllm_causallms,
    vllm_vlms,

--- a/lm_eval/models/rwkv.py
+++ b/lm_eval/models/rwkv.py
@@ -8,7 +8,7 @@ from lm_eval.models.huggingface import HFLM


 @register_model("rwkv")
-class MambaLMWrapper(HFLM):
+class RWKVWRAPPER(HFLM):
    def __init__(
        self,
        pretrained,