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:05:16 -04:00
parent a91a259293
commit e6f1c3ef4c
3 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
replicate/cog.yaml → cog.yaml vendored
View File

5
replicate/predict.py
View File

@ -15,8 +15,11 @@ class Predictor(BasePredictor):
seed: int = Input( seed: int = Input(
description="Specify a random seed.", description="Specify a random seed.",
), ),
grid_size: int = Input(
description="Specify the grid size.",
)
) -> Path: ) -> Path:
image = self.model.generate_image(text, seed, grid_size=3) image = self.model.generate_image(text, seed, grid_size=grid_size)
out_path = Path(tempfile.mkdtemp()) / "output.png" out_path = Path(tempfile.mkdtemp()) / "output.png"
image.save(str(out_path)) image.save(str(out_path))

2
setup.py
View File

@ -5,7 +5,7 @@ setuptools.setup(
name='min-dalle', name='min-dalle',
description = 'min(DALL·E)', description = 'min(DALL·E)',
long_description=(Path(__file__).parent / "README").read_text(), long_description=(Path(__file__).parent / "README").read_text(),
version='0.2.9', version='0.2.10',
author='Brett Kuprel', author='Brett Kuprel',
author_email='brkuprel@gmail.com', author_email='brkuprel@gmail.com',
url='https://github.com/kuprel/min-dalle', url='https://github.com/kuprel/min-dalle',