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