Skip to content

vllm_omni.diffusion.models.seedvr2.config ¶

Operator limits and colour options owned by SeedVR2.

COLOR_CORRECTION_METHODS `module-attribute` ¶

COLOR_CORRECTION_METHODS = (
    "lab",
    "wavelet",
    "adain",
    "none",
)

DEFAULT_COLOR_CORRECTION_METHOD `module-attribute` ¶

DEFAULT_COLOR_CORRECTION_METHOD = 'lab'

environment_variables `module-attribute` ¶

environment_variables: dict[str, Callable[[], Any]] = {
    "VLLM_OMNI_SEEDVR2_MAX_FRAMES": lambda: _positive_int(
        "VLLM_OMNI_SEEDVR2_MAX_FRAMES", 257
    ),
    "VLLM_OMNI_SEEDVR2_SHARDED_FRAME_PIXELS": lambda: (
        _positive_int(
            "VLLM_OMNI_SEEDVR2_SHARDED_FRAME_PIXELS",
            2560 * 1472,
        )
    ),
    "VLLM_OMNI_SEEDVR2_SHARDED_CLIP_PIXELS": lambda: (
        _positive_int(
            "VLLM_OMNI_SEEDVR2_SHARDED_CLIP_PIXELS", None
        )
    ),
    "VLLM_OMNI_SEEDVR2_LONG_MAX_UPLOAD_BYTES": lambda: (
        _positive_int(
            "VLLM_OMNI_SEEDVR2_LONG_MAX_UPLOAD_BYTES",
            8 * 1024**3,
        )
    ),
    "VLLM_OMNI_SEEDVR2_LONG_MAX_WINDOW": lambda: (
        _positive_int(
            "VLLM_OMNI_SEEDVR2_LONG_MAX_WINDOW", 121
        )
    ),
    "VLLM_OMNI_SEEDVR2_LONG_JOB_TTL_SECONDS": lambda: (
        _positive_int(
            "VLLM_OMNI_SEEDVR2_LONG_JOB_TTL_SECONDS", 3600
        )
    ),
    "VLLM_OMNI_SEEDVR2_LONG_OUTPUT_DIR": lambda: (
        os.environ.get(
            "VLLM_OMNI_SEEDVR2_LONG_OUTPUT_DIR",
            tempfile.gettempdir(),
        )
    ),
}