Merge branch 'main' of https://github.com/ModelSurge/sd-webui-comfyui …

…into reverse-proxy
ModelSurge · Aug 16, 2023 · 5826ab7 · 5826ab7
2 parents e57f7d0 + e5ef429
commit 5826ab7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib_comfyui/webui/workflow_patcher.py b/lib_comfyui/webui/workflow_patcher.py
@@ -57,7 +57,7 @@ def sample_img2img_hijack(p, x, *args, original_function, **kwargs):
     processed_x = external_code.run_workflow(
         workflow_type=default_workflow_types.preprocess_latent_workflow_type,
         tab='img2img',
-        batch_input=type_conversion.webui_latent_to_comfyui(x).to(device='cpu'),
+        batch_input=type_conversion.webui_latent_to_comfyui(x.to(device='cpu')),
         identity_on_error=True,
     )
     verify_singleton(processed_x)
@@ -97,7 +97,7 @@ def p_img2img_init(*args, original_function, p_ref, **kwargs):
     processed_images = external_code.run_workflow(
         workflow_type=default_workflow_types.preprocess_workflow_type,
         tab='img2img',
-        batch_input=torch.stack([type_conversion.webui_image_to_comfyui(image) for image in p_ref.init_images]),
+        batch_input=type_conversion.webui_image_to_comfyui(p_ref.init_images),
         identity_on_error=True,
     )
     verify_singleton(processed_images)