diff --git a/tests/load_tests/test_proxy_chat_completions_memory_growth.py b/tests/load_tests/test_proxy_chat_completions_memory_growth.py index ddab06555d1..4cb73b1e6d2 100644 --- a/tests/load_tests/test_proxy_chat_completions_memory_growth.py +++ b/tests/load_tests/test_proxy_chat_completions_memory_growth.py @@ -41,7 +41,7 @@ from tests.load_tests.memory_leak_utils import ( ) # Memory limit for all linear memory growth tests -MEMORY_LIMIT = "170 MB" +MEMORY_LIMIT = "45 MB" # Test Configuration TEST_API_KEY = "sk-1234" diff --git a/tests/load_tests/test_router_acompletion_memory_growth.py b/tests/load_tests/test_router_acompletion_memory_growth.py index 33b60d44c95..05d0e7e630a 100644 --- a/tests/load_tests/test_router_acompletion_memory_growth.py +++ b/tests/load_tests/test_router_acompletion_memory_growth.py @@ -29,8 +29,7 @@ from tests.load_tests.memory_leak_utils import ( ) # Memory limit for all linear memory growth tests -# Increased to 60 MB to account for larger batch size (100 vs 20) which creates more temporary objects -MEMORY_LIMIT = "60 MB" +MEMORY_LIMIT = "45 MB" @pytest.mark.asyncio