update replicate max grid size

2 years ago · e8dac15810
parent 31fd7c231d
commit e8dac15810
1 changed files with 2 additions and 2 deletions
--- a/replicate/predict.py
+++ b/replicate/predict.py
@ -20,8 +20,8 @@ class Predictor(BasePredictor):
        grid_size: int = Input(
            description='Grid Size',
            ge=1,
-            le=3,
-            default=3
+            le=2,
+            default=2
        )
    ) -> Path:
        image = self.model.generate_image(text, seed, grid_size=grid_size)