Quickfix Enum Issue 2 (#2849)

Mikubill · May 5, 2024 · 2c59fec · 2c59fec
1 parent 86288de
commit 2c59fec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
@@ -1216,7 +1216,12 @@ def recolor_intensity_post_processing(x, i):
                     weight = param.weight
 
                 h, w, hr_y, hr_x = Script.get_target_dimensions(p)
-                pulid_mode = PuLIDMode(unit.pulid_mode) if isinstance(unit.pulid_mode, str) else unit.pulid_mode
+                # TODO: Fix all enum issue
+                if unit.pulid_mode == "PuLIDMode.FIDELITY":
+                    pulid_mode = PuLIDMode.FIDELITY
+                else:
+                    pulid_mode = PuLIDMode(unit.pulid_mode)
+
                 if pulid_mode == PuLIDMode.STYLE:
                     pulid_attn_setting = PULID_SETTING_STYLE
                 else: