ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-13 03:15:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #974 from brianteeman/repetition_penalty

Browse source 
repetition_penalty
This commit is contained in:
Tal 2024-06-17 08:39:16 +03:00 committed by GitHub
commit 73ec67b14e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View file

@ -64,7 +64,7 @@ class LiteLLMAIHandler(BaseAiHandler):
if get_settings().get("OLLAMA.API_BASE", None):
litellm.api_base = get_settings().ollama.api_base
self.api_base = get_settings().ollama.api_base
if get_settings().get("HUGGINGFACE.REPITITION_PENALTY", None):
if get_settings().get("HUGGINGFACE.REPETITION_PENALTY", None):
self.repetition_penalty = float(get_settings().huggingface.repetition_penalty)
if get_settings().get("VERTEXAI.VERTEX_PROJECT", None):
litellm.vertex_project = get_settings().vertexai.vertex_project