nodes/city96-ComfyUI_NetDist
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add batch override

Browse Source 
Also fixes image combine, might need to re-add the nodes
This commit is contained in:
City
2023-07-30 00:14:36 +02:00
parent d9292fc161
commit 8215a3c6df
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/misc.py
+1 -1
View File
@@ -21,7 +21,7 @@ class CombineImageBatch:
def combine_images(self,images_a,images_b): def combine_images(self,images_a,images_b):
try: try:
out = torch.cat((images_a,images_b)) out = torch.cat((images_a,images_b), 0)
except RuntimeError: except RuntimeError:
print(f"Imagine size mismatch! {images_a.size()}, {images_b.size()}") print(f"Imagine size mismatch! {images_a.size()}, {images_b.size()}")
out = images_a out = images_a
nodes/remote_control.py
+6 -1
View File
@@ -80,6 +80,7 @@ class QueueRemote:
"default": "http://127.0.0.1:8188/", "default": "http://127.0.0.1:8188/",
}), }),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
"batch_override": ("INT", {"default": 0, "min": 0, "max": 8}),
"system": (["windows", "posix"],), "system": (["windows", "posix"],),
"enabled": (["true", "false", "remote"],{"default": "true"}), "enabled": (["true", "false", "remote"],{"default": "true"}),
"node_id": ("INT", {"default": 1, "min": 1, "max": 64}), "node_id": ("INT", {"default": 1, "min": 1, "max": 64}),
@@ -94,13 +95,17 @@ class QueueRemote:
FUNCTION = "queue_on_remote" FUNCTION = "queue_on_remote"
CATEGORY = "remote" CATEGORY = "remote"
def queue_on_remote(self, remote_url, seed, system, enabled, node_id, prompt): def queue_on_remote(self, remote_url, seed, batch_override, system, enabled, node_id, prompt):
def get_max_batch_size(prompt): def get_max_batch_size(prompt):
bs = 1 bs = 1
for node in prompt.keys(): for node in prompt.keys():
if prompt[node]["class_type"] == "EmptyLatentImage": if prompt[node]["class_type"] == "EmptyLatentImage":
for k,v in prompt[node]["inputs"].items(): for k,v in prompt[node]["inputs"].items():
if k == "batch_size": if k == "batch_size":
if batch_override > 0:
bs = batch_override
prompt[node]["inputs"]["batch_size"] = batch_override
else:
bs = max(bs,int(v)) bs = max(bs,int(v))
return bs return bs