From 998593ca6594c5cc52c157c93dd4af9ab7a101ec Mon Sep 17 00:00:00 2001 From: Xander Steenbrugge Date: Sun, 5 Jan 2025 22:09:46 +0100 Subject: [PATCH] fix basemodel bug2 --- comfyui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfyui.py b/comfyui.py index bf4bf68..425fb55 100644 --- a/comfyui.py +++ b/comfyui.py @@ -629,7 +629,7 @@ def _validate_comfyui_args(self, workflow, tool): raise Exception(f"Remap parameter {key} is missing original choices: {choices}") def _inject_args_into_workflow(self, workflow, tool, args): - + base_model = "unknown" # Helper function to validate and normalize URLs def validate_url(url): if not isinstance(url, str): @@ -719,7 +719,7 @@ def validate_url(url): # if there's a lora, replace mentions with embedding name if key == "prompt": - if base_model == "flux-dev": + if "flux" in base_model: for lora_key in ["lora", "lora2"]: if args.get(f"use_{lora_key}", False): lora_strength = args.get(f"{lora_key}_strength", 0.7)