EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

close gemma2 in test_verl_engine.py temporarily (#4685)

Browse Source
This commit is contained in:
Yi Zhang
2025-03-23 07:36:46 +08:00
committed by GitHub
parent 8796cebb2c
commit 3c09548d1f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/srt/test_verl_engine.py
View File

@@ -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? # TODO maybe we should add more other models? should we keep it in sync with test_generation_models.py?
CI_MODELS = [ CI_MODELS = [
dict(model_path="meta-llama/Llama-3.1-8B-Instruct"), 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 = [ ALL_OTHER_MODELS = [
dict(model_path="meta-llama/Llama-3.2-1B-Instruct"), dict(model_path="meta-llama/Llama-3.2-1B-Instruct"),