Add a separate preview node to prevent generator from running twice, unique_id is buggy

2023-09-16 22:30:29 +02:00
parent 57bfd8cb13
commit 0ec46fb285
4 changed files with 29 additions and 11 deletions
@@ -1,13 +1,15 @@
-from .nodes import Generator, Loader
+from .nodes import Generator, Loader, Previewer

 NODE_CLASS_MAPPINGS = {
-    "ZuellniExLlamaGenerator": Generator,
    "ZuellniExLlamaLoader": Loader,
+    "ZuellniExLlamaGenerator": Generator,
+    "ZuellniExLlamaPreviewer": Previewer,
 }

 NODE_DISPLAY_NAME_MAPPINGS = {
-    "ZuellniExLlamaGenerator": "ExLlama Generator",
    "ZuellniExLlamaLoader": "ExLlama Loader",
+    "ZuellniExLlamaGenerator": "ExLlama Generator",
+    "ZuellniExLlamaPreviewer": "ExLlama Previewer",
 }

 WEB_DIRECTORY = "."