File size: 2,523 Bytes
f545704 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | """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",
]
|