Quickfix Enum Issue 2 (#2849)

2024-05-05 10:07:04 -04:00 · 2024-05-05 10:07:04 -04:00 · 2c59fec402
parent 86288def48
commit 2c59fec402
1 changed files with 6 additions and 1 deletions
--- a/scripts/controlnet.py
+++ b/scripts/controlnet.py
@ -1216,7 +1216,12 @@ class Script(scripts.Script, metaclass=(
                    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: