handle log2_mid_count=0
This commit is contained in:
parent
1702d3c439
commit
97d22f0fdd
|
@ -206,7 +206,7 @@ class MinDalle:
|
|||
attention_state,
|
||||
image_tokens
|
||||
)
|
||||
if handle_intermediate_image is not None:
|
||||
if handle_intermediate_image is not None and log2_mid_count > 0:
|
||||
if ((row_index + 1) * (2 ** log2_mid_count)) % row_count == 0:
|
||||
tokens = image_tokens[:, 1:]
|
||||
image = self.image_from_tokens(grid_size, tokens, is_verbose)
|
||||
|
|
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.22',
|
||||
version='0.2.23',
|
||||
author='Brett Kuprel',
|
||||
author_email='brkuprel@gmail.com',
|
||||
url='https://github.com/kuprel/min-dalle',
|
||||
|
|
Loading…
Reference in New Issue
Block a user