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

update replicate file

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-02 09:20:42 -04:00
parent e087c53482
commit 72cd1501f2
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

@ -13,10 +13,10 @@ class Predictor(BasePredictor):
description="Text for generating images.", description="Text for generating images.",
), ),
seed: int = Input( seed: int = Input(
description="Specify the seed.", description="Specify a random seed.",
), ),
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed) image = self.model.generate_image(text, seed, grid_size=3)
out_path = Path(tempfile.mkdtemp()) / "output.png" out_path = Path(tempfile.mkdtemp()) / "output.png"
image.save(str(out_path)) image.save(str(out_path))