tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

is_reusable

Browse Source
This commit is contained in:
Brett Kuprel
2022-06-30 11:25:24 -04:00
parent 3e64e868ef
commit f951424e38
4 changed files with 64 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files
image_from_text.py
+3 -3
View File
@@ -44,9 +44,9 @@ def generate_image(
image_path: str,
token_count: int
):
is_expendable = True
is_reusable = False
if is_torch:
image_generator = MinDalleTorch(is_mega, is_expendable, token_count)
image_generator = MinDalleTorch(is_mega, is_reusable, token_count)
if token_count < image_generator.config['image_length']:
image_tokens = image_generator.generate_image_tokens(text, seed)
@@ -56,7 +56,7 @@ def generate_image(
image = image_generator.generate_image(text, seed)
else:
image_generator = MinDalleFlax(is_mega, is_expendable=True)
image_generator = MinDalleFlax(is_mega, is_reusable)
image = image_generator.generate_image(text, seed)
save_image(image, image_path)