diff --git a/__init__.py b/__init__.py index 5f97e56..e07eda9 100644 --- a/__init__.py +++ b/__init__.py @@ -315,6 +315,29 @@ class VixUiWorkflowMetadata: return (text,) +class VixDynamicLoraDefinition: + @classmethod + def INPUT_TYPES(cls): + return { + "required": { + "model": ("MODEL", {"tooltip": "The diffusion model the LoRA will be applied to."}), + "clip": ("CLIP", {"tooltip": "The CLIP model the LoRA will be applied to."}), + "base_model_type": ("STRING", {"tooltip": "The base type of model in CivitAI format."}), + "description": ("STRING", {"tooltip": "Brief explanation of LoRA functionality at the added place."}), + }, + } + + RETURN_TYPES = ("MODEL", "CLIP") + OUTPUT_TOOLTIPS = ("The modified diffusion model.", "The modified CLIP model.") + CATEGORY = "Visionatrix/UI" + FUNCTION = "do_it" + DESCRIPTION = "Node that allows dynamic selection of any supported LoRAs from CivitAI in the Visionatrix UI." + + @classmethod + def do_it(cls, model, clip) -> tuple: + return model, clip + + NODE_CLASS_MAPPINGS = { "VixUiCheckbox": VixUiCheckbox, "VixUiRangeFloat": VixUiRangeFloat, @@ -325,6 +348,7 @@ NODE_CLASS_MAPPINGS = { "VixUiCheckboxLogic": VixUiCheckboxLogic, "VixUiListLogic": VixUiListLogic, "VixUiWorkflowMetadata": VixUiWorkflowMetadata, + "VixDynamicLoraDefinition": VixDynamicLoraDefinition, } NODE_DISPLAY_NAME_MAPPINGS = { @@ -337,4 +361,5 @@ NODE_DISPLAY_NAME_MAPPINGS = { "VixUiCheckboxLogic": "VixUI-CheckboxLogic", "VixUiListLogic": "VixUI-ListLogic", "VixUiWorkflowMetadata": "VixUI-WorkflowMetadata", + "VixDynamicLoraDefinition": "VixUI-VixDynamicLoraDefinition", } diff --git a/pyproject.toml b/pyproject.toml index ec8b37b..222fe88 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-visionatrix" description = "The ComfyUI-Visionatrix nodes are designed for convenient ComfyUI to [a/Visionatrix](https://github.com/Visionatrix/Visionatrix) workflow support migration, in particular to extract prompt input params (input, textarea, checkbox, select, range, file) to be used in simplified Visionatrix UI." -version = "1.0.1" +version = "1.0.2" license = {file = "LICENSE"} urls.Repository = "https://github.com/Visionatrix/ComfyUI-Visionatrix"