Skip to content

GitLab

Commit 7d286ad0 authored by Baber's avatar Baber
Browse files

add model

parent 24214440
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
lm_eval/models/__init__.py
View file @ 7d286ad0
...@@ -13,6 +13,7 @@ from . import ( ...@@ -13,6 +13,7 @@ from . import (
openai_completions, openai_completions,
optimum_ipex, optimum_ipex,
optimum_lm, optimum_lm,
rwkvwrapper,
textsynth, textsynth,
vllm_causallms, vllm_causallms,
vllm_vlms, vllm_vlms,
......
lm_eval/models/rwkv.py → lm_eval/models/rwkvwrapper.py
View file @ 7d286ad0
...@@ -8,7 +8,7 @@ from lm_eval.models.huggingface import HFLM ...@@ -8,7 +8,7 @@ from lm_eval.models.huggingface import HFLM
@register_model("rwkv") @register_model("rwkv")
class MambaLMWrapper(HFLM): class RWKVWRAPPER(HFLM):
def __init__( def __init__(
self, self,
pretrained, pretrained,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment