update replicate files
This commit is contained in:
parent
a91a259293
commit
e6f1c3ef4c
0
replicate/cog.yaml → cog.yaml
vendored
0
replicate/cog.yaml → cog.yaml
vendored
|
@ -15,8 +15,11 @@ class Predictor(BasePredictor):
|
|||
seed: int = Input(
|
||||
description="Specify a random seed.",
|
||||
),
|
||||
grid_size: int = Input(
|
||||
description="Specify the grid size.",
|
||||
)
|
||||
) -> 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"
|
||||
image.save(str(out_path))
|
||||
|
||||
|
|
2
setup.py
2
setup.py
|
@ -5,7 +5,7 @@ setuptools.setup(
|
|||
name='min-dalle',
|
||||
description = 'min(DALL·E)',
|
||||
long_description=(Path(__file__).parent / "README").read_text(),
|
||||
version='0.2.9',
|
||||
version='0.2.10',
|
||||
author='Brett Kuprel',
|
||||
author_email='brkuprel@gmail.com',
|
||||
url='https://github.com/kuprel/min-dalle',
|
||||
|
|
Loading…
Reference in New Issue
Block a user