diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py
index d7cfb8962a..2e5140e0e3 100644
--- a/litellm/proxy/auth/user_api_key_auth.py
+++ b/litellm/proxy/auth/user_api_key_auth.py
@@ -1907,6 +1907,7 @@ async def _run_centralized_common_checks(
     skip_budget_checks = _should_skip_budget_checks(
         request_data=request_data,
         route=route,
+        request=request,
         llm_router=llm_router,
     )
 
@@ -1988,9 +1989,14 @@ async def _reserve_budget_after_common_checks(
 def _should_skip_budget_checks(
     request_data: dict,
     route: str,
+    request: Optional[Request],
     llm_router: Optional[Any],
 ) -> bool:
-    model = get_model_from_request(request_data, route)
+    model = _get_model_from_request_context(
+        request_data=request_data,
+        route=route,
+        request=request,
+    )
     if model is not None and llm_router is not None:
         return _is_model_cost_zero(model=model, llm_router=llm_router)
     return False