fix inference category for face fusion

This commit is contained in:
Brandon Thomas committed 2024-09-05 12:41:18 -04:00
1 parent 292ff5d970
commit 0663fd77d1
3 files changed
+5 -2

No files matched your search

@@ -18,10 +18,11 @@ use utoipa::ToSchema;
pub enum InferenceCategory {
/// Deprecate this field !!!
/// We should drain all jobs from using this database field, then remove it.
#[deprecated(note = "NB(bt,2024-09-05): The frontend still needs this")]
#[serde(rename = "deprecated_field")]
DeprecatedField,
/// Facial lipsync animation (eg. SadTalker, Wav2Lip)
/// Facial lipsync animation (eg. SadTalker, Wav2Lip, FaceFusion)
#[serde(rename = "lipsync_animation")]
#[default]
LipsyncAnimation,
@@ -26,7 +26,9 @@ pub struct InsertGenericInferenceArgs<'a> {
pub maybe_product_category: Option<InferenceJobProductCategory>,
// TODO(bt,2024-09-05): We really need to kill this, but the frontend depends on it.
pub inference_category: InferenceCategory,
pub maybe_model_type: Option<InferenceModelType>,
pub maybe_model_token: Option<&'a str>,
@@ -250,7 +250,7 @@ pub async fn enqueue_face_fusion_workflow_handler(
uuid_idempotency_token: &request.uuid_idempotency_token,
job_type: InferenceJobType::FaceFusion,
maybe_product_category: Some(InferenceJobProductCategory::VidFaceFusion),
inference_category: InferenceCategory::DeprecatedField,
inference_category: InferenceCategory::LipsyncAnimation,
maybe_model_type: None,
maybe_model_token: None,
maybe_input_source_token: None,