feat(nodes): add controlnet nodes type hint

2024-08-30 20:32:17 +00:00 · 2023-07-08 19:31:32 +10:00 · 2023-07-08 19:31:32 +10:00 · 82fa39b531
commit 82fa39b531
parent 788dcbde70
1 changed files with 1 additions and 0 deletions
--- a/invokeai/app/invocations/controlnet_image_processors.py
+++ b/invokeai/app/invocations/controlnet_image_processors.py
@ -141,6 +141,7 @@ class ControlField(BaseModel):
            "ui": {
                "type_hints": {
                    "control_weight": "float",
+                    "control_model": "controlnet_model",
                    # "control_weight": "number",
                }
            }