Skip to content

GitLab

Unverified Commit 361a7463 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

fix m2 test (#27536) 


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent 720af6ab
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tests/models/registry.py
View file @ 361a7463
...@@ -342,7 +342,9 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -342,7 +342,9 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
"MiniMaxAI/MiniMax-M1-40k", trust_remote_code=True "MiniMaxAI/MiniMax-M1-40k", trust_remote_code=True
), ),
"MiniMaxM2ForCausalLM": _HfExamplesInfo( "MiniMaxM2ForCausalLM": _HfExamplesInfo(
"MiniMaxAI/MiniMax-M2", trust_remote_code=True "MiniMaxAI/MiniMax-M2",
trust_remote_code=True,
is_available_online=False,
), ),
"MistralForCausalLM": _HfExamplesInfo("mistralai/Mistral-7B-Instruct-v0.1"), "MistralForCausalLM": _HfExamplesInfo("mistralai/Mistral-7B-Instruct-v0.1"),
"MixtralForCausalLM": _HfExamplesInfo( "MixtralForCausalLM": _HfExamplesInfo(
......
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