Karl/ai-frame-image-server
1
0
Fork 0
mirror of https://github.com/karl0ss/ai_image_frame_server.git synced 2025-04-29 03:33:39 +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

3
lib.py
View File

@ -200,10 +200,9 @@ 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)
)