Fix issue with custom nodes that don't have required inputs #266

Acly · Dec 23, 2023 · 3c3bd77 · 3c3bd77
1 parent 09e5ac8
commit 3c3bd77
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ai_diffusion/__init__.py b/ai_diffusion/__init__.py
@@ -1,6 +1,6 @@
 """Generative AI plugin for Krita using Stable Diffusion"""
 
-__version__ = "1.11.0"
+__version__ = "1.11.1"
 
 import importlib.util
 

diff --git a/ai_diffusion/client.py b/ai_diffusion/client.py
@@ -147,7 +147,7 @@ async def connect(url=default_url):
         ]
         if len(missing) > 0:
             raise MissingResource(ResourceKind.node, missing)
-        client.nodes_inputs = {name: nodes[name]["input"]["required"] for name in nodes}
+        client.nodes_inputs = {name: nodes[name]["input"].get("required", None) for name in nodes}
 
         # Retrieve list of checkpoints
         client._refresh_models(nodes, await client.try_inspect_checkpoints())