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

3
lib.py
View File

@ -200,10 +200,9 @@ 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( valid_models = list(
set(get_available_models()) & set(available_model_list) set(get_available_models()) & set(available_model_list)
) )