From b347f9ae63909e1e5c92a46d82abdd41e6b96175 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 1 Feb 2024 16:25:32 -0800
Subject: [PATCH] Create controlnet_unit.py

---
 .../lib_controlnet/controlnet_unit.py         | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_unit.py

diff --git a/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_unit.py b/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_unit.py
new file mode 100644
index 00000000..3da30a72
--- /dev/null
+++ b/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_unit.py
@@ -0,0 +1,19 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class ControlNetUnitGradioState:
+    enabled: bool = True
+    preprocessor_name: str = "None"
+    control_model_name: str = "None"
+    weight: float = 1.0
+    image = None
+    mask = None
+    resize_mode: str = "Crop and Resize"
+    processor_resolution: int = -1
+    slider_1: float = -1
+    slider_2: float = -1
+    guidance_start: float = 0.0
+    guidance_end: float = 1.0
+    pixel_perfect: bool = False
+    control_mode: str = "Balanced"