diff --git a/__init__.py b/__init__.py index c9fc7b4..1acb654 100644 --- a/__init__.py +++ b/__init__.py @@ -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"