Skip to content
Discussion options

You must be logged in to vote

if you wish to pass multiple input images the image count and the "batch_size" needs to be equal
so if you have 2 input images you need batch size of 2

elif len(imgs) <= self.batch_size:
self.batch_size = len(imgs)
batch_images = np.array(imgs)
else:
raise RuntimeError(f"bad number of images passed: {len(imgs)}; expecting {self.batch_size} or less")

note
it also works if the batch size is greater than the image count, as it is automatically set to the number of the input image
but that could be potentially other processes of webui or extensions that already uses the initial ba…

Replies: 1 comment 4 replies

Comment options

You must be logged in to vote
4 replies
@Faishun
Comment options

@w-e-w
Comment options

w-e-w Apr 10, 2024
Collaborator

@Faishun
Comment options

@w-e-w
Comment options

w-e-w Apr 10, 2024
Collaborator

Answer selected by Faishun
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants