Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Load image batches #707

Closed
wants to merge 11 commits into from
+1,023 −344
Closed

Load image batches #707

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
d934119
Image multi upload widget
space-nuko May 29, 2023
e4be8e0
Allow specifying which inputs should be used as lists
space-nuko May 29, 2023
4110e62
Fix drag and drop
space-nuko May 29, 2023
f9063bd
Multiselect image dialog
space-nuko May 29, 2023
569c226
Fix dialog Z order
space-nuko May 29, 2023
c5ea042
`torch.cat()` everything
space-nuko May 29, 2023
2c146b8
Fix litegraph dialog font family
space-nuko May 29, 2023
ec9ff00
Adjust batch image prompt
space-nuko Jun 1, 2023
341fe22
Merge LoadImageBatch into LoadImage
space-nuko Jun 2, 2023
d4e7387
Support previously saved workflows
space-nuko Jun 2, 2023
ec8678c
Load image batch improvements
space-nuko Jun 9, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 3 additions & 6 deletions comfy_extras/nodes_rebatch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,10 @@
class LatentRebatch:
@classmethod
def INPUT_TYPES(s):
return {"required": { "latents": ("LATENT",),
return {"required": { "latents": ("LATENT", { "is_list": True }),
"batch_size": ("INT", {"default": 1, "min": 1, "max": 64}),
}}
RETURN_TYPES = ("LATENT",)
INPUT_IS_LIST = True
RETURN_TYPES = ("LATENT", )
OUTPUT_IS_LIST = (True, )

FUNCTION = "rebatch"
Expand Down Expand Up @@ -54,8 +53,6 @@ def cat_batch(batch1, batch2):
return result

def rebatch(self, latents, batch_size):
batch_size = batch_size[0]

output_list = []
current_batch = (None, None, None)
processed = 0
Expand Down Expand Up @@ -105,4 +102,4 @@ def rebatch(self, latents, batch_size):

NODE_DISPLAY_NAME_MAPPINGS = {
"RebatchLatents": "Rebatch Latents",
}
}
Loading