diff --git a/index.ts b/index.ts
index d667e1a..5c824c6 100644
--- a/index.ts
+++ b/index.ts
@@ -531,6 +531,7 @@ getBucketLifecycleConfiguration on bucket ${this.options.s3Bucket} in region ${t
                 prompt,
                 inputFiles: attachmentFiles,
                 n: 1,
+                size: this.options.generation.outputSize,
               }
             )
 
diff --git a/types.ts b/types.ts
index a0c369a..e2e2d22 100644
--- a/types.ts
+++ b/types.ts
@@ -118,6 +118,10 @@ export type PluginOptions = {
   generation?: {
     adapter: ImageGenerationAdapter,
 
+    /**
+     * The size of the generated image.
+     */
+    outputSize?: string,
 
     /**
      * Fields for conetext which will be used to generate the image.