Set default value of ControlNetUnit.enabled to True (#2912)

Mikubill · May 20, 2024 · a5457dc · a5457dc
1 parent b197f7c
commit a5457dc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/internal_controlnet/args.py b/internal_controlnet/args.py
@@ -73,7 +73,7 @@ class Config:
     loopback: bool = False
 
     # General fields.
-    enabled: bool = False
+    enabled: bool = True
     module: str = "none"
 
     @validator("module", always=True, pre=True)

diff --git a/scripts/controlnet_ui/controlnet_ui_group.py b/scripts/controlnet_ui/controlnet_ui_group.py
@@ -666,7 +666,7 @@ def render(self, tabname: str, elem_id_tabname: str) -> None:
             self.pulid_mode,
         )
 
-        unit = gr.State(ControlNetUnit())
+        unit = gr.State(ControlNetUnit(enabled=False))
 
         # It is necessary to update unit state actively to avoid potential
         # flaky racing issue.

diff --git a/tests/web_api/detect_test.py b/tests/web_api/detect_test.py
@@ -94,6 +94,7 @@ def test_inpaint_mask(module: str):
 
 @disable_in_cq
 @pytest.mark.parametrize("img_index", [i for i, _ in enumerate(portrait_imgs)])
+@pytest.mark.skip(reason="test is for SDXL only")
 def test_pulid(img_index: int):
     """PuLID preprocessor should not memory leak."""
     payload = dict(

diff --git a/tests/web_api/generation_test.py b/tests/web_api/generation_test.py
@@ -310,6 +310,7 @@ def test_ip_adapter_auto():
 
 @disable_in_cq
 @pytest.mark.parametrize("img_index", [i for i, _ in enumerate(portrait_imgs)])
+@pytest.mark.skip(reason="test is for SDXL only")
 def test_pulid(img_index: int):
     """PuLID should not memory leak."""
     assert APITestTemplate(