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

update replicate

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-02 16:36:51 -04:00
parent 9e146e6675
commit 9bdba57195
1 changed files with 2 additions and 2 deletions
Show Stats 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=2, le=3,
default=2 default=3
) )
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed, grid_size=grid_size) image = self.model.generate_image(text, seed, grid_size=grid_size)