Karl/ai-frame-image-server
1
0
Fork 0
mirror of https://github.com/karl0ss/ai_image_frame_server.git synced 2025-04-28 19:23:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed logic

Browse Source
This commit is contained in:
Karl Hudgell 2025-04-21 19:37:20 +01:00
parent eb59cfa25e
commit 3e974d5d79
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib.py
View File

@ -200,16 +200,15 @@ def generate_image(
# Conditionally set model if node and param are provided # Conditionally set model if node and param are provided
if model_node and model_param: if model_node and model_param:
if user_config["comfyui"].get("FLUX"): if user_config["comfyui"].get("FLUX"):
available_model_list = user_config["comfyui:flux"]["models"].split(",") valid_models = user_config["comfyui:flux"]["models"].split(",")
else: else:
available_model_list = user_config["comfyui"]["models"].split(",") available_model_list = user_config["comfyui"]["models"].split(",")
valid_models = list(
set(get_available_models()) & set(available_model_list)
)
valid_models = list( if not valid_models:
set(get_available_models()) & set(available_model_list) raise Exception("No valid models available.")
)
if not valid_models:
raise Exception("No valid models available.")
model = random.choice(valid_models) model = random.choice(valid_models)
wf.set_node_param(model_node, model_param, model) wf.set_node_param(model_node, model_param, model)