mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-11 12:23:20 +00:00
add model to model list
This commit is contained in:
1 parent
4681a3fe56
commit
1f5584f9b5
2 files changed
+28
No files matched your search
@@ -19,6 +19,33 @@ pub const OMNI_GEN_VIDEO_MODELS_AND_PROVIDERS: Lazy<OmniGenVideoModelsResponse>
|
||||
fn build_omni_gen_video_models() -> Vec<OmniGenVideoModelDetails> {
|
||||
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),
|
||||
|
||||
+1
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user