update cog

This commit is contained in:
Brett Kuprel 2022-07-04 20:10:37 -04:00
parent 18b2c84a57
commit 92b4dc3333
3 changed files with 4 additions and 4 deletions

2
cog.yaml vendored
View File

@ -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

View File

@ -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

View File

@ -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',