Update __init__.py

2023-11-05 10:16:41 -08:00
parent 592bece6a4
commit cf30135b83
1 changed files with 2 additions and 1 deletions
@@ -1,4 +1,4 @@
-from .nodes import IntLiteral, FloatLiteral, StringLiteral, CheckpointListLiteral
+from .nodes import IntLiteral, FloatLiteral, StringLiteral, CheckpointListLiteral, LoraListLiteral
 from .operations import Operation
 from .startup_utils import symlink_web_dir

@@ -8,6 +8,7 @@ NODE_CLASS_MAPPINGS = {
    "String": StringLiteral,
    "Operation": Operation,
    "Checkpoint": CheckpointListLiteral,
+    "Lora": LoraListLiteral,
 }

 EXTENSION_NAME = "ComfyLiterals"