[CI/Build] Update bamba test model location (#18544)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
2025-05-22 15:01:07 +02:00
parent a35a494745
commit ca86a7cf6e
3 changed files with 3 additions and 3 deletions
--- a/tests/models/language/generation/test_hybrid.py
+++ b/tests/models/language/generation/test_hybrid.py
@@ -31,7 +31,7 @@ HYBRID_MODELS = [
    # not compatible with pip-compile.
    "pfnet/plamo-2-1b",
    "Zyphra/Zamba2-1.2B-instruct",
-    "hmellor/bamba-tiny-random",
+    "hmellor/tiny-random-BambaForCausalLM",
 ]

 # Avoid OOM