update model

morsecodist · Nov 25, 2023 · 5683957 · 5683957
1 parent 3d3538e
commit 5683957
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -21,8 +21,8 @@
 
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 
-MODEL = "gpt-3.5-turbo-16k"
-TOKENS = 16_385
+MODEL = "gpt-4-32k"
+TOKENS = 32_768
 
 
 def truncate_text(text):