diff --git a/comfy-nodes/external_lora.py b/comfy-nodes/external_lora.py index 8a0322b..c230fd7 100644 --- a/comfy-nodes/external_lora.py +++ b/comfy-nodes/external_lora.py @@ -37,6 +37,10 @@ class ComfyUIDeployExternalLora: "STRING", {"multiline": True, "default": ""}, ), + "lora_url": ( + "STRING", + {"multiline": False, "default": ""}, + ), }, } @@ -47,12 +51,12 @@ class ComfyUIDeployExternalLora: CATEGORY = "deploy" - def run(self, input_id, default_lora_name=None, lora_save_name=None, display_name=None, description=None): + def run(self, input_id, default_lora_name=None, lora_save_name=None, display_name=None, description=None, lora_url=None): import requests import os import uuid - if default_lora_name.startswith("http"): + if lora_url and lora_url.startswith("http"): if lora_save_name: existing_loras = folder_paths.get_filename_list("loras") # Check if lora_save_name exists in the list diff --git a/custom_routes.py b/custom_routes.py index 42f5a25..ecf9e67 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -289,7 +289,7 @@ def apply_inputs_to_workflow(workflow_api: Any, inputs: Any, sid: str = None): value['inputs']["images"] = new_value if value["class_type"] == "ComfyUIDeployExternalLora": - value["inputs"]["default_lora_name"] = new_value + value["inputs"]["lora_url"] = new_value if value["class_type"] == "ComfyUIDeployExternalSlider": value["inputs"]["default_value"] = new_value