From 900bb9aba80f2279fdcd03e8918d45402d4e1d35 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 15 May 2024 07:12:43 -0700 Subject: [PATCH] test(test_token_counter.py): fix load test --- litellm/tests/test_token_counter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_token_counter.py b/litellm/tests/test_token_counter.py index 4d851573cc..78e276a85c 100644 --- a/litellm/tests/test_token_counter.py +++ b/litellm/tests/test_token_counter.py @@ -178,7 +178,7 @@ def test_load_test_token_counter(model): messages = [{"role": "user", "content": text}] * 10 start_time = time.time() - for _ in range(50): + for _ in range(10): _ = token_counter(model=model, messages=messages) # enc.encode("".join(m["content"] for m in messages))