VoxtralForConditionalGeneration(
(audio_tower): VoxtralEncoder(
(conv1): Conv1d(128, 1280, kernel_size=(3,), stride=(1,), padding=(1,))
(conv2): Conv1d(1280, 1280, kernel_size=(3,), stride=(2,), padding=(1,))
(embed_positions): Embedding(1500, 1280)
(layers): ModuleList(
(0-31): 32 x VoxtralEncoderLayer(
(self_attn): VoxtralAttention(
(k_proj): Linear(in_features=1280, out_features=1280, bias=False)
(v_proj): Linear(in_features=1280, out_features=1280, bias=True)
(q_proj): Linear(in_features=1280, out_features=1280, bias=True)
(out_proj): Linear(in_features=1280, out_features=1280, bias=True)
)
(self_attn_layer_norm): LayerNorm((1280,), eps=1e-05, elementwise_affine=True)
(activation_fn): GELUActivation()
(fc1): Linear(in_features=1280, out_features=5120, bias=True)
(fc2): Linear(in_features=5120, out_features=1280, bias=True)
(final_layer_norm): LayerNorm((1280,), eps=1e-05, elementwise_affine=True)
)
)
(layer_norm): LayerNorm((1280,), eps=1e-05, elementwise_affine=True)
(avg_pooler): AvgPool1d(kernel_size=(2,), stride=(2,), padding=(0,))
)
(language_model): LlamaForCausalLM(
(model): LlamaModel(
(embed_tokens): Embedding(131072, 5120)
(layers): ModuleList(
(0-39): 40 x LlamaDecoderLayer(
(self_attn): LlamaAttention(
(q_proj): Linear(in_features=5120, out_features=4096, bias=False)
(k_proj): Linear(in_features=5120, out_features=1024, bias=False)
(v_proj): Linear(in_features=5120, out_features=1024, bias=False)
(o_proj): Linear(in_features=4096, out_features=5120, bias=False)
)
(mlp): LlamaMLP(
(gate_proj): Linear(in_features=5120, out_features=32768, bias=False)
(up_proj): Linear(in_features=5120, out_features=32768, bias=False)
(down_proj): Linear(in_features=32768, out_features=5120, bias=False)
(act_fn): SiLU()
)
(input_layernorm): LlamaRMSNorm((5120,), eps=1e-05)
(post_attention_layernorm): LlamaRMSNorm((5120,), eps=1e-05)
)
)
(norm): LlamaRMSNorm((5120,), eps=1e-05)
(rotary_emb): LlamaRotaryEmbedding()
)
(lm_head): Linear(in_features=5120, out_features=131072, bias=False)
)
(multi_modal_projector): VoxtralMultiModalProjector(
(linear_1): Linear(in_features=5120, out_features=5120, bias=False)
(act): GELUActivation()
(linear_2): Linear(in_features=5120, out_features=5120, bias=False)
)
)
模型架构如上,devide_map如下:
OrderedDict([('audio_tower', 2), ('language_model.model.embed_tokens', 2), ('language_model.model.layers.0', 2), ('language_model.model.layers.1', 2), ('language_model.model.layers.2', 2), ('language_model.model.layers.3', 2), ('language_model.model.layers.4', 2), ('language_model.model.layers.5', 2), ('language_model.model.layers.6', 2), ('language_model.model.layers.7', 2), ('language_model.model.layers.8', 2), ('language_model.model.layers.9', 2), ('language_model.model.layers.10', 2), ('language_model.model.layers.11', 2), ('language_model.model.layers.12', 2), ('language_model.model.layers.13', 2), ('language_model.model.layers.14', 2), ('language_model.model.layers.15', 2), ('language_model.model.layers.16', 2), ('language_model.model.layers.17', 2), ('language_model.model.layers.18', 2), ('language_model.model.layers.19', 2), ('language_model.model.layers.20', 2), ('language_model.model.layers.21', 2), ('language_model.model.layers.22', 2), ('language_model.model.layers.23', 2), ('language_model.model.layers.24', 2), ('language_model.model.layers.25.self_attn', 2), ('language_model.model.layers.25.input_layernorm', 5), ('language_model.model.layers.25.post_attention_layernorm', 5), ('language_model.model.layers.26', 5), ('language_model.model.layers.27', 5), ('language_model.model.layers.28', 5), ('language_model.model.layers.29', 5), ('language_model.model.layers.30', 5), ('language_model.model.layers.31', 5), ('language_model.model.layers.32', 5), ('language_model.model.layers.33', 5), ('language_model.model.layers.34', 5), ('language_model.model.layers.35', 5), ('language_model.model.layers.36', 5), ('language_model.model.layers.37', 5), ('language_model.model.layers.38', 5), ('language_model.model.layers.39', 5), ('language_model.model.norm', 5), ('language_model.model.rotary_emb', 5), ('language_model.lm_head', 5), ('multi_modal_projector', 5), ('language_model.model.layers.25.mlp', 5)])
运行报错,分析一下