update cog
This commit is contained in:
2
cog.yaml
vendored
2
cog.yaml
vendored
@@ -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
|
||||
|
||||
|
@@ -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
|
||||
|
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.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',
|
||||
|
Reference in New Issue
Block a user