Skip to content

GitLab

Unverified Commit ab9f2cfd authored by Thomas Parnell's avatar Thomas Parnell Committed by GitHub
Browse files

[CI] [Hybrid] Bump min transformers version for Bamba and Jamba (#22908) 


Signed-off-by: default avatarThomas Parnell <tpa@zurich.ibm.com>
parent dbe29804
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/registry.py
View file @ ab9f2cfd
...@@ -151,7 +151,7 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -151,7 +151,7 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
"BailingMoeForCausalLM": _HfExamplesInfo("inclusionAI/Ling-lite-1.5", "BailingMoeForCausalLM": _HfExamplesInfo("inclusionAI/Ling-lite-1.5",
trust_remote_code=True), trust_remote_code=True),
"BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B-v1", "BambaForCausalLM": _HfExamplesInfo("ibm-ai-platform/Bamba-9B-v1",
min_transformers_version="4.55.1", min_transformers_version="4.56.0",
extras={"tiny": "hmellor/tiny-random-BambaForCausalLM"}), # noqa: E501 extras={"tiny": "hmellor/tiny-random-BambaForCausalLM"}), # noqa: E501
"BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m", "BloomForCausalLM": _HfExamplesInfo("bigscience/bloom-560m",
{"1b": "bigscience/bloomz-1b1"}), {"1b": "bigscience/bloomz-1b1"}),
...@@ -227,7 +227,7 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -227,7 +227,7 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
trust_remote_code=True), trust_remote_code=True),
"JAISLMHeadModel": _HfExamplesInfo("inceptionai/jais-13b-chat"), "JAISLMHeadModel": _HfExamplesInfo("inceptionai/jais-13b-chat"),
"JambaForCausalLM": _HfExamplesInfo("ai21labs/AI21-Jamba-1.5-Mini", "JambaForCausalLM": _HfExamplesInfo("ai21labs/AI21-Jamba-1.5-Mini",
min_transformers_version="4.55.1", min_transformers_version="4.56.0",
extras={ extras={
"tiny": "ai21labs/Jamba-tiny-dev", "tiny": "ai21labs/Jamba-tiny-dev",
"random": "ai21labs/Jamba-tiny-random", # noqa: E501 "random": "ai21labs/Jamba-tiny-random", # noqa: E501
......
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