Unverified Commit c329ceca authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[CI Fix] Pin tests/models/registry.py MiniMaxText01ForCausalLM to revision due...


[CI Fix] Pin tests/models/registry.py MiniMaxText01ForCausalLM to revision due to model changes (#20199)
Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 3c545c0c
...@@ -70,6 +70,12 @@ class _HfExamplesInfo: ...@@ -70,6 +70,12 @@ class _HfExamplesInfo:
length that is too large to fit into memory in CI. length that is too large to fit into memory in CI.
""" """
revision: Optional[str] = None
"""
The specific revision (commit hash, tag, or branch) to use for the model.
If not specified, the default revision will be used.
"""
def check_transformers_version( def check_transformers_version(
self, self,
*, *,
...@@ -207,7 +213,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -207,7 +213,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
"MiniCPM3ForCausalLM": _HfExamplesInfo("openbmb/MiniCPM3-4B", "MiniCPM3ForCausalLM": _HfExamplesInfo("openbmb/MiniCPM3-4B",
trust_remote_code=True), trust_remote_code=True),
"MiniMaxText01ForCausalLM": _HfExamplesInfo("MiniMaxAI/MiniMax-Text-01", "MiniMaxText01ForCausalLM": _HfExamplesInfo("MiniMaxAI/MiniMax-Text-01",
trust_remote_code=True), trust_remote_code=True,
revision="a59aa9cbc53b9fb8742ca4e9e1531b9802b6fdc3"), # noqa: E501
"MiniMaxM1ForCausalLM": _HfExamplesInfo("MiniMaxAI/MiniMax-M1-40k", "MiniMaxM1ForCausalLM": _HfExamplesInfo("MiniMaxAI/MiniMax-M1-40k",
trust_remote_code=True), trust_remote_code=True),
"MistralForCausalLM": _HfExamplesInfo("mistralai/Mistral-7B-Instruct-v0.1"), "MistralForCausalLM": _HfExamplesInfo("mistralai/Mistral-7B-Instruct-v0.1"),
......
...@@ -88,6 +88,7 @@ def test_can_initialize(model_arch: str, monkeypatch: pytest.MonkeyPatch): ...@@ -88,6 +88,7 @@ def test_can_initialize(model_arch: str, monkeypatch: pytest.MonkeyPatch):
model_info.default, model_info.default,
tokenizer=model_info.tokenizer, tokenizer=model_info.tokenizer,
tokenizer_mode=model_info.tokenizer_mode, tokenizer_mode=model_info.tokenizer_mode,
revision=model_info.revision,
speculative_config={ speculative_config={
"model": model_info.speculative_model, "model": model_info.speculative_model,
"num_speculative_tokens": 1, "num_speculative_tokens": 1,
......
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