[CLEANUP]

The-Swarm-Corporation · Mar 21, 2024 · 935d860 · 935d860
1 parent 214a4ef
commit 935d860
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/servers/cogvlm/cogvlm.py b/servers/cogvlm/cogvlm.py
@@ -163,8 +163,8 @@ async def create_chat_completion(
     entry = ModelAPILogEntry(
         user_id=await fetch_api_key_info(token),
         model_id="41a2869c-5f8d-403f-83bb-1f06c56bad47",
-        input_tokens=await int(count_tokens(request.messages, tokenizer, request.model)),
-        output_tokens=await int(count_tokens(response["text"], tokenizer, request.model)),
+        input_tokens=await count_tokens(request.messages, tokenizer, request.model),
+        output_tokens=await count_tokens(response["text"], tokenizer, request.model),
         all_cost=await calculate_pricing(
             texts=[message.content], tokenizer=tokenizer, rate_per_million=15.0
         ),