Skip to content

Commit 8ff0691

Browse files
committed
Added to supported_models.md and fixed URLs
Signed-off-by: Stanislaw Wozniak <[email protected]>
1 parent 0937f79 commit 8ff0691

File tree

4 files changed

+11
-6
lines changed

4 files changed

+11
-6
lines changed

docs/source/models/supported_models.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -385,6 +385,11 @@ See [this page](#generative-models) for more information on how to use generativ
385385
* `ibm-granite/granite-3.0-1b-a400m-base`, `ibm-granite/granite-3.0-3b-a800m-instruct`, `ibm/PowerMoE-3b`, etc.
386386
* ✅︎
387387
* ✅︎
388+
- * `GraniteMoeHybridForCausalLM`
389+
* Granite 4.0 MoE Hybrid
390+
* `ibm-granite/granite-4.0-tiny-preview`, etc.
391+
* ✅︎
392+
* ✅︎
388393
- * `GraniteMoeSharedForCausalLM`
389394
* Granite MoE Shared
390395
* `ibm-research/moe-7b-1b-active-shared-experts` (test model)

tests/models/language/generation/test_granitemoehybrid.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66

77
# Path of the checkpoints
88
MODELS = [
9-
"ibm-research/granite-4.0-tiny-test",
9+
"ibm-granite/granite-4.0-tiny-preview",
1010
]
1111

1212

13-
@pytest.mark.skip(reason="HF model URLs not available yet")
13+
@pytest.mark.skip(reason="HF model is in the HF main yet")
1414
@pytest.mark.parametrize("model", MODELS)
1515
@pytest.mark.parametrize("dtype", ["float16", "bfloat16"])
1616
@pytest.mark.parametrize("max_tokens", [64])

tests/models/language/generation/test_hybrid.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@
2323

2424
HYBRID_MODELS = [
2525
"ai21labs/Jamba-tiny-dev",
26-
# NOTE: ibm-research/granite-4.0-tiny-test are skipped currently as
27-
# the HF model URLs not available yet
28-
# "ibm-research/granite-4.0-tiny-test",
26+
# NOTE: ibm-granite/granite-4.0-tiny-preview are skipped currently as
27+
# the HF model is in the HF main yet
28+
# "ibm-granite/granite-4.0-tiny-preview",
2929
# NOTE: Running Plamo2 in transformers implementation requires to install
3030
# causal-conv1d package, which is not listed as a test dependency as it's
3131
# not compatible with pip-compile.

tests/models/registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def check_available_online(
163163
{"1b": "EleutherAI/pythia-1.4b"}),
164164
"GraniteForCausalLM": _HfExamplesInfo("ibm/PowerLM-3b"),
165165
"GraniteMoeForCausalLM": _HfExamplesInfo("ibm/PowerMoE-3b"),
166-
"GraniteMoeHybridForCausalLM": _HfExamplesInfo("ibm-research/granite-4.0-tiny-test"), # noqa: E501
166+
"GraniteMoeHybridForCausalLM": _HfExamplesInfo("ibm-granite/granite-4.0-tiny-preview"), # noqa: E501
167167
"GraniteMoeSharedForCausalLM": _HfExamplesInfo("ibm-research/moe-7b-1b-active-shared-experts"), # noqa: E501
168168
"Grok1ModelForCausalLM": _HfExamplesInfo("hpcai-tech/grok-1",
169169
trust_remote_code=True),

0 commit comments

Comments
 (0)