diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 1a0f89bf5..13f880d53 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -149,7 +149,6 @@ suites = { TestFile("test_dp_attention.py", 350), TestFile("test_load_weights_from_remote_instance.py", 72), TestFile("test_patch_torch.py", 19), - TestFile("test_release_memory_occupation.py", 200), TestFile("test_eagle_dp_attention.py", 200), ], "per-commit-4-gpu": [ @@ -203,6 +202,7 @@ suites = { ], # Nightly test suites have been moved to test/run_suite_nightly.py "__not_in_ci__": [ + TestFile("test_release_memory_occupation.py", 200), # Temporarily disabled TestFile("models/test_dummy_grok_models.py"), TestFile( "rl/test_update_weights_from_disk.py"