Draft change to only use ai-binary for code generation, not code fixing

google · Mar 29, 2024 · 749306b · 749306b
1 parent b4928d2
commit 749306b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llm_toolkit/code_fixer.py b/llm_toolkit/code_fixer.py
@@ -334,8 +334,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,
   )