diff --git a/vibe_bot/llama_wrapper.py b/vibe_bot/llama_wrapper.py
index 7a89868..77cef23 100644
--- a/vibe_bot/llama_wrapper.py
+++ b/vibe_bot/llama_wrapper.py
@@ -318,7 +318,7 @@ def image_generation(
         response = client.images.generate(
             prompt=prompt,
             n=n,
-            size="1024x1024",
+            size="768x768",
             model=model,
             timeout=120.0,
         )
@@ -357,7 +357,7 @@ def image_edit(
         image=image,
         prompt=prompt,
         n=n,
-        size="1024x1024",
+        size="768x768",
         model=model,
     )
     if response.data: