added is_verbose flag

This commit is contained in:
Brett Kuprel
2022-07-01 20:17:20 -04:00
parent 35e97768a5
commit cf9656baa2
3 changed files with 23 additions and 19 deletions
+2 -1
View File
@@ -44,7 +44,8 @@ def generate_image(
is_mega=is_mega,
models_root='pretrained',
is_reusable=False,
sample_token_count=token_count
sample_token_count=token_count,
is_verbose=True
)
if token_count < 256: