close gemma2 in test_verl_engine.py temporarily (#4685)

3c09548d · Yi Zhang · GitHub · 8796cebb · 3c09548d
Unverified Commit 3c09548d authored Mar 23, 2025 by Yi Zhang Committed by GitHub Mar 22, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/srt/test_verl_engine.py test/srt/test_verl_engine.py +2 -1

No files found.
--- a/test/srt/test_verl_engine.py
+++ b/test/srt/test_verl_engine.py
@@ -40,7 +40,8 @@ _ENABLE_UPDATE_WEIGHTS = True
 # TODO maybe we should add more other models? should we keep it in sync with test_generation_models.py?
 CI_MODELS = [
    dict(model_path="meta-llama/Llama-3.1-8B-Instruct"),
-    dict(model_path="google/gemma-2-2b"),
+    # Fail to run gemma-2-2b after transformers==4.48.3 -> 4.50.0
+    # dict(model_path="google/gemma-2-2b"),
 ]
 ALL_OTHER_MODELS = [
    dict(model_path="meta-llama/Llama-3.2-1B-Instruct"),