diff --git a/cog.yaml b/cog.yaml index af25f52..ad2cda4 100644 --- a/cog.yaml +++ b/cog.yaml @@ -6,7 +6,7 @@ build: - "libgl1-mesa-glx" - "libglib2.0-0" python_packages: - - "min-dalle==0.2.23" + - "min-dalle==0.2.24" run: - pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html diff --git a/cogrun.py b/cogrun.py index 2da19ff..cfbaf2a 100644 --- a/cogrun.py +++ b/cogrun.py @@ -32,8 +32,8 @@ class Predictor(BasePredictor): ), ) -> Iterator[Path]: image_stream = self.model.generate_image_stream( - text, - seed, + text, + seed, grid_size=grid_size, log2_mid_count=log2_intermediate_image_count, is_verbose=True diff --git a/setup.py b/setup.py index c326523..e6baede 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setuptools.setup( name='min-dalle', description = 'min(DALLĀ·E)', long_description=(Path(__file__).parent / "README.rst").read_text(), - version='0.2.23', + version='0.2.24', author='Brett Kuprel', author_email='brkuprel@gmail.com', url='https://github.com/kuprel/min-dalle',