diff --git a/crates/service/web/storyteller_web/src/configs/omni_gen/video_models.rs b/crates/service/web/storyteller_web/src/configs/omni_gen/video_models.rs index 5de222271a..ad8a7af34d 100644 --- a/crates/service/web/storyteller_web/src/configs/omni_gen/video_models.rs +++ b/crates/service/web/storyteller_web/src/configs/omni_gen/video_models.rs @@ -19,6 +19,33 @@ pub const OMNI_GEN_VIDEO_MODELS_AND_PROVIDERS: Lazy fn build_omni_gen_video_models() -> Vec { let mut models = Vec::new(); + models.push(OmniGenVideoModelDetails { + model: CommonVideoModel::HappyHorse1p0, + model_creator: Some(ModelCreator::Alibaba), + full_name: Some("Happy Horse 1.0".to_string()), + text_prompt_supported: Some(true), + starting_keyframe_supported: Some(true), + aspect_ratio_options: Some(vec![ + CommonAspectRatio::WideSixteenByNine, + CommonAspectRatio::WideFourByThree, + CommonAspectRatio::Square, + CommonAspectRatio::TallThreeByFour, + CommonAspectRatio::TallNineBySixteen, + ]), + aspect_ratio_default: Some(CommonAspectRatio::WideSixteenByNine), + resolution_options: Some(vec![ + CommonResolution::SevenTwentyP, + CommonResolution::TenEightyP, + ]), + resolution_default: Some(CommonResolution::SevenTwentyP), + batch_size_options: Some(vec![1, 2, 4]), + batch_size_default: Some(1), + duration_seconds_min: Some(3), + duration_seconds_max: Some(15), + duration_seconds_default: Some(5), + ..Default::default() + }); + models.push(OmniGenVideoModelDetails { model: CommonVideoModel::Seedance1p5Pro, model_creator: Some(ModelCreator::Bytedance), diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs index f0454c886f..4508b8c9c4 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs @@ -169,6 +169,7 @@ pub async fn omni_gen_video_generate_handler( // ==================== PIPELINE DISPATCH ==================== // let use_v2 = match request.model { + Some(CommonVideoModel::HappyHorse1p0) => true, Some(CommonVideoModel::Seedance2p0) => true, Some(CommonVideoModel::Seedance2p0Fast) => true, _ => false,