Commit 24214440 authored by Baber's avatar Baber
Browse files

nit

parent 6cfc0f28
...@@ -54,8 +54,7 @@ class MambaLMWrapper(HFLM): ...@@ -54,8 +54,7 @@ class MambaLMWrapper(HFLM):
from rwkv.model import RWKV from rwkv.model import RWKV
except ModuleNotFoundError as exception: except ModuleNotFoundError as exception:
raise type(exception)( raise type(exception)(
"attempted to use 'mamba_ssm' LM type, but package `mamba_ssm` is not installed. \ "install rwkv package (pip install rwkv)",
please install mamba via `pip install lm-eval[mamba]` or `pip install -e .[mamba]`",
) )
import os import os
......
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