remove support for batch_size from dream.py (#227)

* remove dream.py support for batch_size

* expect to get a single image
This commit is contained in:
Kevin Gibbons
2022-08-30 19:30:12 -07:00
committed by GitHub
parent b5565d2c82
commit 1714816fe2
6 changed files with 27 additions and 56 deletions

View File

@ -199,7 +199,7 @@ def main_loop(t2i, outdir, prompt_as_dir, parser, infile):
# Here is where the images are actually generated!
try:
file_writer = PngWriter(current_outdir, normalized_prompt, opt.batch_size)
file_writer = PngWriter(current_outdir, normalized_prompt)
callback = file_writer.write_image if individual_images else None
image_list = t2i.prompt2image(image_callback=callback, **vars(opt))
results = (
@ -419,13 +419,6 @@ def create_cmd_parser():
default=1,
help='Number of samplings to perform (slower, but will provide seeds for individual images)',
)
parser.add_argument(
'-b',
'--batch_size',
type=int,
default=1,
help='Number of images to produce per sampling (will not provide seeds for individual images!)',
)
parser.add_argument(
'-W', '--width', type=int, help='Image width, multiple of 64'
)