# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from dataclasses import dataclass

from fastvideo.api.sampling_param import SamplingParam


@dataclass
class FluxSamplingParam(SamplingParam):

    prompt: str | None = "a photo of a cat"
    negative_prompt: str = ""

    num_videos_per_prompt: int = 1
    seed: int = 0

    num_frames: int = 1
    height: int = 1024
    width: int = 1024
    fps: int = 1

    num_inference_steps: int = 28
    guidance_scale: float = 3.5
    use_embedded_guidance: bool = True
    true_cfg_scale: float = 1.0