From 0555e4f2d15bdb7d40b596dedd29975f29b95ef5 Mon Sep 17 00:00:00 2001 From: "Jun Siang Cheah (aider)" Date: Sun, 1 Sep 2024 23:24:30 +0100 Subject: [PATCH] refactor: remove script_dir param from get_groq_limits_for_stt_model --- src/pull_available_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pull_available_models.py b/src/pull_available_models.py index 33bd93b..522f68d 100644 --- a/src/pull_available_models.py +++ b/src/pull_available_models.py @@ -113,7 +113,7 @@ def get_model_name(id): return id -def get_groq_limits_for_stt_model(model_id, script_dir): +def get_groq_limits_for_stt_model(model_id): print(f"Getting limits for STT model {model_id}...") r = requests.post( "https://api.groq.com/openai/v1/audio/transcriptions", @@ -138,7 +138,7 @@ def get_groq_limits_for_stt_model(model_id, script_dir): def get_groq_limits_for_model(model_id, script_dir): if "whisper" in model_id: - return get_groq_limits_for_stt_model(model_id, script_dir) + return get_groq_limits_for_stt_model(model_id) print(f"Getting limits for chat model {model_id}...") r = requests.post( "https://api.groq.com/openai/v1/chat/completions",