From ecc1534f878ceaf4c66181e95be2fda02866b02c Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Mon, 4 Jul 2022 10:28:47 -0400 Subject: [PATCH] 5x5 grid works on colab P100 so should work on replicate --- replicate/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(