update cog
This commit is contained in:
parent
18b2c84a57
commit
92b4dc3333
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',
|
||||
|
|
Loading…
Reference in New Issue
Block a user