From 3e974d5d798fd11d408b958613a9fed9b7c66fd4 Mon Sep 17 00:00:00 2001 From: Karl Hudgell Date: Mon, 21 Apr 2025 19:37:20 +0100 Subject: [PATCH] fixed logic --- lib.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/lib.py b/lib.py index ba77ed6..e660750 100644 --- a/lib.py +++ b/lib.py @@ -200,16 +200,15 @@ def generate_image( # Conditionally set model if node and param are provided if model_node and model_param: if user_config["comfyui"].get("FLUX"): - available_model_list = user_config["comfyui:flux"]["models"].split(",") + valid_models = user_config["comfyui:flux"]["models"].split(",") else: available_model_list = user_config["comfyui"]["models"].split(",") + valid_models = list( + set(get_available_models()) & set(available_model_list) + ) - valid_models = list( - set(get_available_models()) & set(available_model_list) - ) - - if not valid_models: - raise Exception("No valid models available.") + if not valid_models: + raise Exception("No valid models available.") model = random.choice(valid_models) wf.set_node_param(model_node, model_param, model)