Add force offload node

2024-08-02 15:27:46 +02:00
parent 6fb8649168
commit 7e96bf676c
3 changed files with 92 additions and 0 deletions
@@ -0,0 +1,11 @@
+NODE_CLASS_MAPPINGS = {}
+
+from .offload import NODE_CLASS_MAPPINGS as Offload_Nodes
+NODE_CLASS_MAPPINGS.update(Offload_Nodes)
+
+for name, node in NODE_CLASS_MAPPINGS.items():
+	cat = node.CATEGORY
+	if not cat.startswith("ExtraModels/"):
+		node.CATEGORY = f"ExtraModels/{cat}"
+
+__all__ = ["NODE_CLASS_MAPPINGS"]