Split nodes into files, allow loading each separately without cloning the repo, some other minor stuff, install exllamav2 pip package by default

2023-10-07 10:56:18 +02:00
parent 796a6b8f25
commit 6e78068c1a
7 changed files with 198 additions and 180 deletions
@@ -1,17 +1,9 @@
-from .nodes import Generator, Loader, Previewer, Replacer
-
-NODE_CLASS_MAPPINGS = {
-    "ZuellniExLlamaLoader": Loader,
-    "ZuellniExLlamaGenerator": Generator,
-    "ZuellniTextPreviewer": Previewer,
-    "ZuellniTextReplacer": Replacer,
-}
-
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "ZuellniExLlamaLoader": "ExLlama Loader",
-    "ZuellniExLlamaGenerator": "ExLlama Generator",
-    "ZuellniTextPreviewer": "Preview Text",
-    "ZuellniTextReplacer": "Replace Text",
-}
+from . import exllama, text

+NODE_CLASS_MAPPINGS = {}
+NODE_DISPLAY_NAME_MAPPINGS = {}
 WEB_DIRECTORY = "."
+
+for module in (exllama, text):
+    NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
+    NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)