Quick fix enum issue

Mikubill · May 4, 2024 · 49e1bed · 49e1bed
1 parent d702830
commit 49e1bed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
@@ -1209,7 +1209,7 @@ 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)
+                pulid_mode = PuLIDMode(unit.pulid_mode) if isinstance(unit.pulid_mode, str) else unit.pulid_mode
                 if pulid_mode == PuLIDMode.STYLE:
                     pulid_attn_setting = PULID_SETTING_STYLE
                 else: