diff --git a/.ci/scripts/gather_test_models.py b/.ci/scripts/gather_test_models.py index 515bc97cca3..d02213b9faf 100755 --- a/.ci/scripts/gather_test_models.py +++ b/.ci/scripts/gather_test_models.py @@ -90,7 +90,7 @@ def model_should_run_on_event(model: str, event: str) -> bool: We put higher priority and fast models to pull request and rest to push. """ if event == "pull_request": - return model in ["mv3", "vit", "qwen2_5"] # TODO: remove, just to test the ci + return model in ["mv3", "vit"] elif event == "push": # These are super slow. Only run it periodically return model not in ["dl3", "edsr", "emformer_predict"]