| """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", |
| ] |
|
|
