diff --git a/.env.example b/.env.example
index c0caaa3..7520f91 100644
--- a/.env.example
+++ b/.env.example
@@ -37,7 +37,7 @@ ALLOWED_TELEGRAM_USER_IDS=USER_ID_1,USER_ID_2
 # IMAGE_MODEL=dall-e-3
 # IMAGE_QUALITY=hd
 # IMAGE_STYLE=natural
-# IMAGE_SIZE=512x512
+# IMAGE_SIZE=1024x1024
 # GROUP_TRIGGER_KEYWORD=""
 # IGNORE_GROUP_TRANSCRIPTIONS=true
 # BOT_LANGUAGE=en
\ No newline at end of file
diff --git a/bot/openai_helper.py b/bot/openai_helper.py
index a31ac3d..26fad44 100644
--- a/bot/openai_helper.py
+++ b/bot/openai_helper.py
@@ -34,6 +34,8 @@ def default_max_tokens(model: str) -> int:
     """
     base = 1200
     if model in GPT_3_MODELS:
+        if model == "gpt-3.5-turbo-0301":
+            return 4096
         return base
     elif model in GPT_4_MODELS:
         return base * 2
@@ -42,7 +44,7 @@ def default_max_tokens(model: str) -> int:
     elif model in GPT_4_32K_MODELS:
         return base * 8
     elif model in GPT_4_128K_MODELS:
-        return base * 31
+        return 4096
 
 
 def are_functions_available(model: str) -> bool: