Karl/ai-frame-image-server
1
0
Fork 0
mirror of https://github.com/karl0ss/ai-frame-image-server.git synced 2025-04-26 01:30:13 +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 305b55eaf6
commit 9c606451c2
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
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)