tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update replicate max grid size

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-02 11:59:26 -04:00
parent 31fd7c231d
commit e8dac15810
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
replicate/predict.py
View File

@@ -20,8 +20,8 @@ class Predictor(BasePredictor):
grid_size: int = Input( grid_size: int = Input(
description='Grid Size', description='Grid Size',
ge=1, ge=1,
le=3, le=2,
default=3 default=2
) )
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed, grid_size=grid_size) image = self.model.generate_image(text, seed, grid_size=grid_size)