Bump transfomers to 4.54.1 to fix Gemma cache issue. (#8541)

2025-07-30 19:50:54 -07:00
parent 9b9e82539b
commit 67e53b16f5
2 changed files with 2 additions and 3 deletions
--- a/test/srt/models/test_generation_models.py
+++ b/test/srt/models/test_generation_models.py
@@ -51,8 +51,7 @@ class ModelCase:
 # Popular models that run on the CI
 CI_MODELS = [
    ModelCase("meta-llama/Llama-3.1-8B-Instruct"),
-    # TODO: Gemma is broken by the bug introduced in the latest transformers version, we should restore once its fixed: https://github.com/huggingface/transformers/issues/39711
-    # ModelCase("google/gemma-2-2b"),
+    ModelCase("google/gemma-2-2b"),
 ]

 # the complete set of models to test sglang's generation model