Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 0 additions & 4 deletions backend/src/common/base_dto.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,10 +68,6 @@ class GenerationModelEnum(str, Enum):
VTO = "virtual-try-on-001"

# Video-Specific Models
VEO_3_1_FAST_GENERATE_001 = "veo-3.1-fast-generate-001"
VEO_3_1_LITE_GENERATE_001 = "veo-3.1-lite-generate-001"
VEO_3_1_GENERATE_001 = "veo-3.1-generate-001"
VEO_3_1_PREVIEW = "veo-3.1-generate-preview"
VEO_3_FAST = "veo-3.0-fast-generate-001"
VEO_3_QUALITY = "veo-3.0-generate-001"
VEO_3_FAST_PREVIEW = "veo-3.0-fast-generate-preview"
Expand Down
15 changes: 2 additions & 13 deletions backend/src/videos/dto/create_veo_dto.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ class CreateVeoDto(BaseDto):
description="The ID of the workspace for this generation.",
)
generation_model: GenerationModelEnum = Field(
default=GenerationModelEnum.VEO_3_1_GENERATE_001,
default=GenerationModelEnum.VEO_3_QUALITY,
description="Model used for image generation.",
)
aspect_ratio: AspectRatioEnum = Field(
Expand Down Expand Up @@ -167,17 +167,10 @@ def validate_cross_fields(self) -> "CreateVeoDto":
if has_any_references:
if (
model != GenerationModelEnum.VEO_2_GENERATE_EXP
and model != GenerationModelEnum.VEO_3_1_PREVIEW
and model != GenerationModelEnum.VEO_3_1_GENERATE_001
and model != GenerationModelEnum.VEO_3_1_LITE_GENERATE_001
and model != GenerationModelEnum.VEO_3_1_FAST_GENERATE_001
):
raise ValueError(
"Reference images are only supported by the "
f"'{GenerationModelEnum.VEO_3_1_PREVIEW.value}' model, "
f"'{GenerationModelEnum.VEO_3_1_GENERATE_001.value}' model, "
f"'{GenerationModelEnum.VEO_3_1_LITE_GENERATE_001.value}' model, or "
f"'{GenerationModelEnum.VEO_3_1_FAST_GENERATE_001.value}' model.",
f"'{GenerationModelEnum.VEO_2_GENERATE_EXP.value}' model.",
)

start_image_present = bool(self.start_image_asset_id)
Expand Down Expand Up @@ -218,10 +211,6 @@ def validate_video_generation_model(
) -> GenerationModelEnum:
"""Ensures that only supported generation models for video are used."""
valid_video_ratios = [
GenerationModelEnum.VEO_3_1_PREVIEW,
GenerationModelEnum.VEO_3_1_GENERATE_001,
GenerationModelEnum.VEO_3_1_LITE_GENERATE_001,
GenerationModelEnum.VEO_3_1_FAST_GENERATE_001,
GenerationModelEnum.VEO_3_FAST,
GenerationModelEnum.VEO_3_QUALITY,
GenerationModelEnum.VEO_3_FAST_PREVIEW,
Expand Down
Loading