From e45433755ec846c1d13c24e6bab5fc439d489669 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Wed, 7 Feb 2024 22:18:46 -0800 Subject: [PATCH] Include missing seed parameter in sample workflow --- comfy/client/sdxl_with_refiner_workflow.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comfy/client/sdxl_with_refiner_workflow.py b/comfy/client/sdxl_with_refiner_workflow.py index 5706dd633..2b069243a 100644 --- a/comfy/client/sdxl_with_refiner_workflow.py +++ b/comfy/client/sdxl_with_refiner_workflow.py @@ -160,7 +160,8 @@ def sdxl_workflow_with_refiner(prompt: str, height=1024, sampler="euler_ancestral", scheduler="normal", - filename_prefix="sdxl_") -> PromptDict: + filename_prefix="sdxl_", + seed=42) -> PromptDict: prompt_dict: JSON = copy.deepcopy(_BASE_PROMPT) prompt_dict["17"]["inputs"]["text"] = prompt prompt_dict["20"]["inputs"]["text"] = negative_prompt @@ -171,6 +172,7 @@ def sdxl_workflow_with_refiner(prompt: str, # base prompt_dict["10"]["inputs"]["steps"] = inference_steps + refiner_steps + prompt_dict["10"]["inputs"]["seed"] = seed prompt_dict["10"]["inputs"]["start_at_step"] = 0 prompt_dict["10"]["inputs"]["end_at_step"] = inference_steps prompt_dict["10"]["inputs"]["steps"] = inference_steps + refiner_steps @@ -179,6 +181,7 @@ def sdxl_workflow_with_refiner(prompt: str, # refiner prompt_dict["14"]["inputs"]["steps"] = inference_steps + refiner_steps + prompt_dict["10"]["inputs"]["seed"] = seed prompt_dict["14"]["inputs"]["start_at_step"] = inference_steps prompt_dict["14"]["inputs"]["end_at_step"] = inference_steps + refiner_steps prompt_dict["14"]["inputs"]["sampler_name"] = sampler