Fix controlnet with precessor set to none

ramyma · Sep 15, 2023 · e71b3e3 · e71b3e3
1 parent 81cc9ab
commit e71b3e3
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/lib/ex_sd/comfy_client.ex b/lib/ex_sd/comfy_client.ex
@@ -1072,8 +1072,6 @@ defmodule ExSd.ComfyClient do
           },
           class_type: "Base64ImageInput"
         },
-        # FIXME: handle none
-        "cn#{index}_preprocessor": controlnet_preprocessor(index, %{}, entry.module),
         "cn#{index}_controlnet_loader": %{
           inputs: %{
             control_net_name: entry.model
@@ -1112,13 +1110,25 @@ defmodule ExSd.ComfyClient do
               0
             ],
             image: [
-              "cn#{index}_preprocessor",
+              if(
+                entry.module == "none",
+                do: "cn#{index}_image",
+                else: "cn#{index}_preprocessor"
+              ),
               0
             ]
           },
           class_type: "ControlNetApplyAdvanced"
         }
       })
+      |> Map.merge(
+        if(entry.module == "none",
+          do: %{},
+          else: %{
+            "cn#{index}_preprocessor": controlnet_preprocessor(index, %{}, entry.module)
+          }
+        )
+      )
     end)
   end