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

update replicate files

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-02 10:32:17 -04:00
parent e6f1c3ef4c
commit 9cde5b1a60
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cog.yaml vendored
View File

@ -9,4 +9,4 @@ build:
- "torch==1.10.1" - "torch==1.10.1"
- "flax==0.5.2" - "flax==0.5.2"
predict: "predict.py:Predictor" predict: "replicate/predict.py:Predictor"

4
replicate/predict.py
View File

@ -13,10 +13,12 @@ class Predictor(BasePredictor):
description="Text for generating images.", description="Text for generating images.",
), ),
seed: int = Input( seed: int = Input(
description="Specify a random seed.", description="Specify a random seed."
), ),
grid_size: int = Input( grid_size: int = Input(
description="Specify the grid size.", description="Specify the grid size.",
ge=1,
le=4
) )
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed, grid_size=grid_size) image = self.model.generate_image(text, seed, grid_size=grid_size)