diff --git a/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs b/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs index e88d08ca43..5a3320221f 100644 --- a/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs +++ b/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs @@ -79,6 +79,12 @@ pub struct PromptInnerPayload { #[serde(alias = "travel_prompt")] #[serde(skip_serializing_if = "Option::is_none")] pub travel_prompt: Option, + + /// Frame skip + #[serde(rename = "fs")] // NB: DO NOT CHANGE: IT WILL BREAK MYSQL RECORDS. Renamed to consume fewer bytes. + #[serde(alias = "frame_skip")] + #[serde(skip_serializing_if = "Option::is_none")] + pub frame_skip: Option, } pub struct PromptInnerPayloadBuilder { @@ -95,6 +101,7 @@ pub struct PromptInnerPayloadBuilder { pub upscaler_workflow: Option, pub global_ipa_token: Option, pub travel_prompt: Option, + pub frame_skip: Option, } impl PromptInnerPayloadBuilder { @@ -113,6 +120,7 @@ impl PromptInnerPayloadBuilder { upscaler_workflow: None, global_ipa_token: None, travel_prompt: None, + frame_skip: None, } } @@ -130,6 +138,7 @@ impl PromptInnerPayloadBuilder { && self.upscaler_workflow.is_none() && self.global_ipa_token.is_none() && self.travel_prompt.is_none() + && self.frame_skip.is_none() { return None; } @@ -151,6 +160,7 @@ impl PromptInnerPayloadBuilder { upscaler_workflow: self.upscaler_workflow, global_ipa_token: self.global_ipa_token, travel_prompt: self.travel_prompt, + frame_skip: self.frame_skip, }) } @@ -225,6 +235,10 @@ impl PromptInnerPayloadBuilder { pub fn set_travel_prompt(&mut self, travel_prompt: Option) { self.travel_prompt = travel_prompt; } + + pub fn set_frame_skip(&mut self, frame_skip: Option) { + self.frame_skip = frame_skip; + } } impl PromptInnerPayload{ diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs index 7f3b0dc532..1fc28cb31c 100644 --- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs +++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs @@ -294,6 +294,10 @@ pub async fn validate_and_save_results(args: SaveResultsArgs<'_>) -> Result