fix inference category for face fusion
This commit is contained in:
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,
|
||||
|
||||
+2
@@ -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>,
|
||||
|
||||
|
||||
+1
-1
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user