5x5 grid works on colab P100 so should work on replicate

2022-07-04 10:28:47 -04:00
parent 3dc92b19bd
commit ecc1534f87
1 changed files with 1 additions and 1 deletions
--- 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(