From 68ee673cfb84b3f6621015c6122253afea78b42b Mon Sep 17 00:00:00 2001 From: Dongge Liu Date: Sat, 30 Mar 2024 08:38:03 +1100 Subject: [PATCH] Draft change to only use ai-binary for code generation, not code fixing --- llm_toolkit/code_fixer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llm_toolkit/code_fixer.py b/llm_toolkit/code_fixer.py index 54dd8a9252..95a2e357d3 100755 --- a/llm_toolkit/code_fixer.py +++ b/llm_toolkit/code_fixer.py @@ -337,8 +337,8 @@ def apply_llm_fix(ai_binary: str, temperature: float = 0.4): """Queries LLM to fix the code.""" fixer_model = models.LLM.setup( - ai_binary=ai_binary, - name=fixer_model_name, + ai_binary='', + name=models.DefaultModel.name, num_samples=1, temperature=temperature, )