VST: pass ip adapter token to jobs

This commit is contained in:
Brandon Thomas committed 2024-06-22 16:24:55 -04:00
1 parent cb98541c54
commit 5a18c51c11
4 files changed
+11 -2

No files matched your search

@@ -123,6 +123,12 @@ pub struct WorkflowArgs {
#[serde(skip_serializing_if = "Option::is_none")]
pub negative_prompt: Option<String>,
// Optional global IP Adapter image media file token
// This will apply to the entire workflow / video
#[serde(rename = "gi")]
#[serde(skip_serializing_if = "Option::is_none")]
pub global_ip_adapter_token: Option<MediaFileToken>,
#[deprecated(note = "Use `lipsync_enabled` instead")]
#[serde(rename = "el")]
#[serde(skip_serializing_if = "Option::is_none")]
@@ -277,6 +277,7 @@ pub async fn enqueue_comfy_ui_handler(
trim_end_milliseconds: None,
positive_prompt: None,
negative_prompt: None,
global_ip_adapter_token: None,
enable_lipsync: None,
rollout_python_workflow_args: get_request_header_optional(&http_request, "PYTHON-WORKFLOW-ARGS")
.map(|value| str_to_bool(&value)),
@@ -120,10 +120,9 @@ pub struct EnqueueVideoStyleTransferRequest {
/// Optional visibility setting override.
creator_set_visibility: Option<Visibility>,
// TODO(bt,2024-05-13): Plumb this through to jobs.
/// Optional Global IP-Adapter image.
/// The underlying media file must be an image.
ipa_media_token: Option<MediaFileToken>,
global_ipa_media_token: Option<MediaFileToken>,
}
#[derive(Serialize, ToSchema)]
@@ -354,8 +353,10 @@ pub async fn enqueue_video_style_transfer_handler(
trim_end_milliseconds: Some(trim_end_millis),
positive_prompt: request.prompt.new_string_trim_or_empty(),
negative_prompt: request.negative_prompt.new_string_trim_or_empty(),
global_ip_adapter_token: request.global_ipa_media_token.as_ref().map(|t| t.clone()),
strength: maybe_strength,
// Flags
disable_lcm: coordinated_args.disable_lcm,
use_cinematic: coordinated_args.use_cinematic,
use_face_detailer: coordinated_args.use_face_detailer,
@@ -246,6 +246,7 @@ pub async fn enqueue_workflow_upload_request(
trim_end_milliseconds: None,
positive_prompt: None,
negative_prompt: None,
global_ip_adapter_token: None,
enable_lipsync: None,
rollout_python_workflow_args: get_request_header_optional(&http_request, "PYTHON-WORKFLOW-ARGS")
.map(|value| str_to_bool(&value)),