From 4bc8f117b9bf326e084fb209ba506cf8fc30e6af Mon Sep 17 00:00:00 2001 From: Karl Hudgell Date: Fri, 13 Dec 2024 12:07:28 +0000 Subject: [PATCH] num_inference_steps --- comfy_fm_newgen.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy_fm_newgen.py b/comfy_fm_newgen.py index 6e55698..122d412 100644 --- a/comfy_fm_newgen.py +++ b/comfy_fm_newgen.py @@ -27,7 +27,7 @@ p = inflect.engine() output_folder = user_config["general"]["output_dir"] -def generate_image(uid, comfy_prompt, model): +def generate_image(uid, comfy_prompt, model, steps): """Generate an image using the Comfy API.""" url = user_config["general"]["url"] @@ -37,7 +37,7 @@ def generate_image(uid, comfy_prompt, model): # Set workflow parameters wf.set_node_param("KSampler", "seed", random.getrandbits(32)) - wf.set_node_param("KSampler", "steps", 6) + wf.set_node_param("KSampler", "steps", steps) wf.set_node_param("positive", "text", comfy_prompt) wf.set_node_param("Save Image", "filename_prefix", uid) wf.set_node_param("Load Checkpoint", "ckpt_name", model) @@ -127,6 +127,7 @@ def main(): uid, comfy_prompt, user_config["general"]["model"], + args.num_inference_steps ) remove_bg_from_files_in_dir(output_folder) resize_images(output_folder)