File size: 2,289 Bytes
b0cfefd |
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 |
arch:
type: TransformerLMHeadModel
args:
transformer_config:
type: TransformerDecoderOnlyModel
args:
embed_config:
type: TransformerEmbeddingBlock
args:
token_embed_config:
type: TokenEmbedding
args:
n_embed: 2048
n_vocab: 32000
pos_embed_config: null
type_embed_config: null
ln_config: null
p_drop_embed: 0.0
concat_strategy: id_first
decoder_config:
type: TransformerDecoderBlock
args:
attn_config:
type: LlamaAttention
args:
n_embed: 2048
n_pos: 2048
n_head: 32
n_key_value_head: 4
head_size: 64
p_drop_attn: 0.0
p_drop_resid: 0.0
bias_attn: false
bias_proj: false
cross_attn: false
scale_dot_product: true
scale_layer_wise: false
layer_idx: null
rope_config:
type: RotaryPositionEmbedding
args:
head_size: 64
n_pos: 2048
base: 10000
scaling_type: null
scaling_factor: null
mlp_config:
type: LlamaMLP
args:
n_embed: 2048
n_inner: 5632
act_fn_config:
type: SiLUActivation
args: {}
ln_config:
type: LlamaRMSNorm
args:
n_embed: 2048
ln_eps: 1.0e-05
n_embed: 2048
post_norm: false
add_cross_attn: false
n_embed: 2048
n_layer: 22
n_head: 32
ln_config:
type: LlamaRMSNorm
args:
n_embed: 2048
ln_eps: 1.0e-05
perform_linear_bias: false
attn_window_size_loop_unit: null
lm_head_config:
type: TransformerLMHead
args:
n_vocab: 32000
n_embed: 2048
perform_transform: false
act_fn_config: null
ln_config: null
|