main
Brett Kuprel 2 years ago
parent 18b2c84a57
commit 92b4dc3333
  1. 2
      cog.yaml
  2. 4
      cogrun.py
  3. 2
      setup.py

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

@ -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…
Cancel
Save