diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_inference_command.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_inference_command.rs index e9e1f0ada5..707932c08d 100644 --- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_inference_command.rs +++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_inference_command.rs @@ -67,6 +67,7 @@ pub enum ExecutableOrCommand { pub struct InferenceArgs<'s> { pub stderr_output_file: &'s Path, pub stdout_output_file: &'s Path, + pub prompt_location: &'s Path, } impl ComfyInferenceCommand { @@ -156,6 +157,10 @@ impl ComfyInferenceCommand { } } + command.push_str(" --prompt "); + command.push_str(&path_to_string(args.prompt_location)); + command.push_str(" "); + if let Some(docker_options) = self.maybe_docker_options.as_ref() { command = docker_options.to_command_string(&command); } diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/process_job.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/process_job.rs index 17577cabbe..44a108e91e 100644 --- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/process_job.rs +++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/process_job.rs @@ -324,16 +324,20 @@ pub async fn process_job(args: ComfyProcessJobArgs<'_>) -> Result