display intermediate images

image_from_text.py

@@ -1,7 +1,6 @@
 import argparse
 import os
 from PIL import Image
-
 from min_dalle import MinDalle
 
 
@@ -9,7 +8,7 @@ parser = argparse.ArgumentParser()
 parser.add_argument('--mega', action='store_true')
 parser.add_argument('--no-mega', dest='mega', action='store_false')
 parser.set_defaults(mega=False)
-parser.add_argument('--text', type=str, default='alien life')
+parser.add_argument('--text', type=str, default='Dali painting of WALL·E')
 parser.add_argument('--seed', type=int, default=-1)
 parser.add_argument('--grid-size', type=int, default=1)
 parser.add_argument('--image-path', type=str, default='generated')
@@ -17,7 +16,7 @@ parser.add_argument('--models-root', type=str, default='pretrained')
 parser.add_argument('--row-count', type=int, default=16) # for debugging
 
 
-def ascii_from_image(image: Image.Image, size: int) -> str:
+def ascii_from_image(image: Image.Image, size: int = 128) -> str:
     rgb_pixels = image.resize((size, int(0.55 * size))).convert('L').getdata()
     chars = list('.,;/IOX')
     chars = [chars[i * len(chars) // 256] for i in rgb_pixels]
@@ -57,12 +56,13 @@ def generate_image(
             text, 
             seed, 
             grid_size ** 2, 
-            row_count
+            row_count,
+            is_verbose=True
         )
         image_tokens = image_tokens[:, :token_count].to('cpu').detach().numpy()
         print('image tokens', image_tokens)
     else:
-        image = model.generate_image(text, seed, grid_size)
+        image = model.generate_image(text, seed, grid_size, is_verbose=True)
         save_image(image, image_path)
         print(ascii_from_image(image, size=128))