diff --git a/replicate/predict.py b/replicate/predict.py
index 69c51ea..f407c40 100644
--- a/replicate/predict.py
+++ b/replicate/predict.py
@@ -16,7 +16,7 @@ class Predictor(BasePredictor):
         grid_size: int = Input(
             description='Size of the image grid',
             ge=1,
-            le=4,
+            le=5,
             default=4
         ),
         seed: int = Input(