Fix img2img (#142)

* fix img2img * batches
ModelSurge · Aug 16, 2023 · e5ef429 · e5ef429
1 parent 5d62813
commit e5ef429
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib_comfyui/webui/workflow_patcher.py b/lib_comfyui/webui/workflow_patcher.py
@@ -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)