feat: worker allow_sdxl_controlnet bridge data option

horde_sdk/ai_horde_api/apimodels/generate/_pop.py

@@ -492,6 +492,7 @@ class ImageGenerateJobPopRequest(BaseAIHordeRequest, APIKeyAllowedInRequestMixin
     allow_unsafe_ipaddr: bool = True
     allow_post_processing: bool = True
     allow_controlnet: bool = False
+    allow_sdxl_controlnet: bool = False
     allow_lora: bool = False
 
     @override

horde_sdk/ai_horde_worker/bridge_data.py

@@ -141,6 +141,9 @@ class ImageWorkerBridgeData(SharedHordeBridgeData):
     allow_controlnet: bool = False
     """Whether to allow the use of ControlNet. This requires img2img to be enabled."""
 
+    allow_sdxl_controlnet: bool = False
+    """Whether to allow the use of SDXL ControlNet. This requires controlnet to be enabled."""
+
     allow_img2img: bool = True
     """Whether to allow the use of img2img."""
 
@@ -298,6 +301,15 @@ def validate_model(self) -> ImageWorkerBridgeData:
             )
             self.allow_controlnet = False
 
+        if not self.allow_controlnet and self.allow_sdxl_controlnet:
+            logger.warning(
+                (
+                    "allow_sdxl_controlnet is set to True, but allow_controlnet is set to False. "
+                    "SDXL ControlNet requires allow_controlnet to be enabled. Setting allow_sdxl_controlnet to false."
+                ),
+            )
+            self.allow_sdxl_controlnet = False
+
         self.image_models_to_skip.append("SDXL_beta::stability.ai#6901")  # FIXME: no magic strings
 
         return self