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

randomly select dev/schnell when using flux

Browse Source
This commit is contained in:
Karl Hudgell 2025-04-21 10:31:13 +01:00
parent ab50f2afaf
commit 60be7c4a00
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lib.py
View File

@ -199,15 +199,22 @@ 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"):
available_model_list = user_config["comfyui"]["confyui:flux"]["models"].split(",")
else:
available_model_list = user_config["comfyui"]["models"].split(",")
valid_models = list( valid_models = list(
set(get_available_models()) set(get_available_models()) & set(available_model_list)
& set(user_config["comfyui"]["models"].split(","))
) )
if not valid_models: if not valid_models:
raise Exception("No valid models available.") 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)
# Generate image # Generate image
logging.debug(f"Generating image: {file_name}") logging.debug(f"Generating image: {file_name}")
results = api.queue_and_wait_images(wf, save_node) results = api.queue_and_wait_images(wf, save_node)
@ -244,8 +251,8 @@ def create_image(prompt: str | None = None) -> None:
seed_param="seed", seed_param="seed",
save_node="CivitAI Image Saver", save_node="CivitAI Image Saver",
save_param="filename", save_param="filename",
model_node=None, # FLUX doesn't use model selection model_node="CivitAI Image Saver",
model_param=None, model_param="modelname",
) )
else: else:
generate_image("image", prompt) generate_image("image", prompt)

3
user_config.cfg.sample
View File

@ -13,6 +13,9 @@ width = 1568
height = 672 height = 672
FLUX = False FLUX = False
[comfyui:flux]
models = flux1-dev-Q4_0.gguf,flux1-schnell-Q4_0.gguf
[openwebui] [openwebui]
base_url = https://openwebui base_url = https://openwebui
api_key = sk- api_key = sk-