From d4d1f873e03cfc029d2892390245fc997493a22f Mon Sep 17 00:00:00 2001
From: Dazed <dazedanon@gmail.com>
Date: Sun, 10 Dec 2023 18:46:17 -0600
Subject: [PATCH] Force Frequency Penalty depending on Model

---
 modules/rpgmakermvmz.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/modules/rpgmakermvmz.py b/modules/rpgmakermvmz.py
index 128ee02..2589cf3 100644
--- a/modules/rpgmakermvmz.py
+++ b/modules/rpgmakermvmz.py
@@ -41,10 +41,12 @@ if 'gpt-3.5' in MODEL:
     INPUTAPICOST = .002 
     OUTPUTAPICOST = .002
     BATCHSIZE = 10
+    FREQUENCY_PENALTY = 0.2
 elif 'gpt-4' in MODEL:
     INPUTAPICOST = .01
     OUTPUTAPICOST = .03
     BATCHSIZE = 40  
+    FREQUENCY_PENALTY = 0
 
 #tqdm Globals
 BAR_FORMAT='{l_bar}{bar:10}{r_bar}{bar:-10b}'
@@ -1876,8 +1878,8 @@ def translateText(characters, system, user, history):
     response = openai.chat.completions.create(
         temperature=0.1,
         top_p = 0.2,
-        frequency_penalty=0.1,
-        presence_penalty=0.1,
+        frequency_penalty=FREQUENCY_PENALTY,
+        presence_penalty=FREQUENCY_PENALTY,
         model=MODEL,
         messages=msg,
     )