"""Z-Image Configuration."""

from .inference import (
    DEFAULT_CFG_TRUNCATION,
    DEFAULT_GUIDANCE_SCALE,
    DEFAULT_HEIGHT,
    DEFAULT_INFERENCE_STEPS,
    DEFAULT_MAX_SEQUENCE_LENGTH,
    DEFAULT_WIDTH,
)
from .model import (
    ADALN_EMBED_DIM,
    BASE_IMAGE_SEQ_LEN,
    BASE_SHIFT,
    BYTES_PER_GB,
    DEFAULT_LOAD_DEVICE,
    DEFAULT_LOAD_DTYPE_STR,
    DEFAULT_SCHEDULER_NUM_TRAIN_TIMESTEPS,
    DEFAULT_SCHEDULER_SHIFT,
    DEFAULT_SCHEDULER_USE_DYNAMIC_SHIFTING,
    DEFAULT_TRANSFORMER_CAP_FEAT_DIM,
    DEFAULT_TRANSFORMER_DIM,
    DEFAULT_TRANSFORMER_F_PATCH_SIZE,
    DEFAULT_TRANSFORMER_IN_CHANNELS,
    DEFAULT_TRANSFORMER_N_HEADS,
    DEFAULT_TRANSFORMER_N_KV_HEADS,
    DEFAULT_TRANSFORMER_N_LAYERS,
    DEFAULT_TRANSFORMER_N_REFINER_LAYERS,
    DEFAULT_TRANSFORMER_NORM_EPS,
    DEFAULT_TRANSFORMER_PATCH_SIZE,
    DEFAULT_TRANSFORMER_QK_NORM,
    DEFAULT_TRANSFORMER_T_SCALE,
    DEFAULT_VAE_IN_CHANNELS,
    DEFAULT_VAE_LATENT_CHANNELS,
    DEFAULT_VAE_NORM_NUM_GROUPS,
    DEFAULT_VAE_OUT_CHANNELS,
    DEFAULT_VAE_SCALE_FACTOR,
    DEFAULT_VAE_SCALING_FACTOR,
    FREQUENCY_EMBEDDING_SIZE,
    MAX_IMAGE_SEQ_LEN,
    MAX_PERIOD,
    MAX_SHIFT,
    ROPE_AXES_DIMS,
    ROPE_AXES_LENS,
    ROPE_THETA,
    SEQ_MULTI_OF,
)

__all__ = [
    "ADALN_EMBED_DIM",
    "SEQ_MULTI_OF",
    "ROPE_THETA",
    "ROPE_AXES_DIMS",
    "ROPE_AXES_LENS",
    "FREQUENCY_EMBEDDING_SIZE",
    "MAX_PERIOD",
    "BASE_IMAGE_SEQ_LEN",
    "MAX_IMAGE_SEQ_LEN",
    "BASE_SHIFT",
    "MAX_SHIFT",
    "DEFAULT_VAE_SCALE_FACTOR",
    "DEFAULT_VAE_IN_CHANNELS",
    "DEFAULT_VAE_OUT_CHANNELS",
    "DEFAULT_VAE_LATENT_CHANNELS",
    "DEFAULT_VAE_NORM_NUM_GROUPS",
    "DEFAULT_VAE_SCALING_FACTOR",
    "DEFAULT_TRANSFORMER_PATCH_SIZE",
    "DEFAULT_TRANSFORMER_F_PATCH_SIZE",
    "DEFAULT_TRANSFORMER_IN_CHANNELS",
    "DEFAULT_TRANSFORMER_DIM",
    "DEFAULT_TRANSFORMER_N_LAYERS",
    "DEFAULT_TRANSFORMER_N_REFINER_LAYERS",
    "DEFAULT_TRANSFORMER_N_HEADS",
    "DEFAULT_TRANSFORMER_N_KV_HEADS",
    "DEFAULT_TRANSFORMER_NORM_EPS",
    "DEFAULT_TRANSFORMER_QK_NORM",
    "DEFAULT_TRANSFORMER_CAP_FEAT_DIM",
    "DEFAULT_TRANSFORMER_T_SCALE",
    "DEFAULT_SCHEDULER_NUM_TRAIN_TIMESTEPS",
    "DEFAULT_SCHEDULER_SHIFT",
    "DEFAULT_SCHEDULER_USE_DYNAMIC_SHIFTING",
    "DEFAULT_LOAD_DEVICE",
    "DEFAULT_LOAD_DTYPE_STR",
    "BYTES_PER_GB",
    "DEFAULT_HEIGHT",
    "DEFAULT_WIDTH",
    "DEFAULT_INFERENCE_STEPS",
    "DEFAULT_GUIDANCE_SCALE",
    "DEFAULT_CFG_TRUNCATION",
    "DEFAULT_MAX_SEQUENCE_LENGTH",
]