GitHub
185.1K
Download
Models
Docs
Pricing
Sign in
Download
Models
Docs
Pricing
Sign in
lfm3-omni
:2b-bf16
4,744
Downloads
Updated
20 minutes ago
a single end-to-end model that understands text, images, video and audio, and answers in text and speech
a single end-to-end model that understands text, images, video and audio, and answers in text and speech
Cancel
vision
tools
audio
2b
lfm3-omni:2b-bf16
...
/
projector
92907148363e ยท 950MB
Metadata
general.architecture
clip
clip
general.file_type
BF16
BF16
clip.audio.attention.head_count
8
8
clip.audio.attention.head_count_kv
4
4
clip.audio.attention.layer_norm_epsilon
1e-05
1e-05
clip.audio.block_count
17
17
clip.audio.embedding_length
768
768
clip.audio.feed_forward_length
1792
1792
clip.audio.num_mel_bins
128
128
clip.audio.projection_dim
2048
2048
clip.audio.projector_type
lfm3a
lfm3a
clip.audio.window_size
32
32
clip.gen.audio.attention.head_count
32
32
clip.gen.audio.attention.head_count_kv
8
8
clip.gen.audio.attention.layer_norm_epsilon
1e-05
1e-05
clip.gen.audio.block_count
3
3
clip.gen.audio.embedding_length
1024
1024
clip.gen.audio.feed_forward_length
2816
2816
clip.gen.audio.projection_dim
2048
2048
clip.gen.audio.projector_type
lfm3a_gen
lfm3a_gen
clip.gen.audio.sampling.penalty_last_n
64
64
clip.gen.audio.sampling.penalty_repeat
1.2
1.2
clip.gen.audio.sampling.temp
0.4
0.4
clip.has_audio_encoder
true
true
clip.has_gen_audio_encoder
true
true
clip.has_vision_encoder
true
true
clip.use_silu
true
true
clip.vision.attention.head_count
16
16
clip.vision.attention.head_count_kv
8
8
clip.vision.attention.layer_norm_epsilon
1e-05
1e-05
clip.vision.block_count
14
14
clip.vision.embedding_length
1024
1024
clip.vision.feed_forward_length
2560
2560
clip.vision.image_max_pixels
2097152
2097152
clip.vision.image_mean
[0.5, 0.5, 0.5]
[0.5, 0.5, 0.5]
clip.vision.image_min_pixels
65536
65536
clip.vision.image_resize_algo
bicubic
bicubic
clip.vision.image_size
512
512
clip.vision.image_std
[0.5, 0.5, 0.5]
[0.5, 0.5, 0.5]
clip.vision.patch_size
16
16
clip.vision.projection_dim
2048
2048
clip.vision.projector_type
lfm3vl
lfm3vl
clip.vision.spatial_merge_size
2
2
Tensor
Name
Type
Shape
a.blk.0
a.blk.0.conv_dw.weight
F32
F32
[3, 768]
a.blk.0.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.0.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.0.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.0.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.0.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.0.ln1.weight
F32
F32
[768]
a.blk.0.ln2.weight
F32
F32
[768]
a.blk.1
a.blk.1.conv_dw.weight
F32
F32
[3, 768]
a.blk.1.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.1.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.1.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.1.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.1.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.1.ln1.weight
F32
F32
[768]
a.blk.1.ln2.weight
F32
F32
[768]
a.blk.2
a.blk.2.attn_k.weight
BF16
BF16
[768, 384]
a.blk.2.attn_k_norm.weight
F32
F32
[96]
a.blk.2.attn_out.weight
BF16
BF16
[768, 768]
a.blk.2.attn_q.weight
BF16
BF16
[768, 768]
a.blk.2.attn_q_norm.weight
F32
F32
[96]
a.blk.2.attn_v.weight
BF16
BF16
[768, 384]
a.blk.2.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.2.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.2.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.2.ln1.weight
F32
F32
[768]
a.blk.2.ln2.weight
F32
F32
[768]
a.blk.3
a.blk.3.conv_dw.weight
F32
F32
[3, 768]
a.blk.3.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.3.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.3.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.3.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.3.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.3.ln1.weight
F32
F32
[768]
a.blk.3.ln2.weight
F32
F32
[768]
a.blk.4
a.blk.4.conv_dw.weight
F32
F32
[3, 768]
a.blk.4.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.4.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.4.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.4.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.4.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.4.ln1.weight
F32
F32
[768]
a.blk.4.ln2.weight
F32
F32
[768]
a.blk.5
a.blk.5.attn_k.weight
BF16
BF16
[768, 384]
a.blk.5.attn_k_norm.weight
F32
F32
[96]
a.blk.5.attn_out.weight
BF16
BF16
[768, 768]
a.blk.5.attn_q.weight
BF16
BF16
[768, 768]
a.blk.5.attn_q_norm.weight
F32
F32
[96]
a.blk.5.attn_v.weight
BF16
BF16
[768, 384]
a.blk.5.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.5.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.5.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.5.ln1.weight
F32
F32
[768]
a.blk.5.ln2.weight
F32
F32
[768]
a.blk.6
a.blk.6.conv_dw.weight
F32
F32
[3, 768]
a.blk.6.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.6.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.6.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.6.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.6.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.6.ln1.weight
F32
F32
[768]
a.blk.6.ln2.weight
F32
F32
[768]
a.blk.7
a.blk.7.conv_dw.weight
F32
F32
[3, 768]
a.blk.7.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.7.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.7.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.7.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.7.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.7.ln1.weight
F32
F32
[768]
a.blk.7.ln2.weight
F32
F32
[768]
a.blk.8
a.blk.8.attn_k.weight
BF16
BF16
[768, 384]
a.blk.8.attn_k_norm.weight
F32
F32
[96]
a.blk.8.attn_out.weight
BF16
BF16
[768, 768]
a.blk.8.attn_q.weight
BF16
BF16
[768, 768]
a.blk.8.attn_q_norm.weight
F32
F32
[96]
a.blk.8.attn_v.weight
BF16
BF16
[768, 384]
a.blk.8.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.8.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.8.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.8.ln1.weight
F32
F32
[768]
a.blk.8.ln2.weight
F32
F32
[768]
a.blk.9
a.blk.9.conv_dw.weight
F32
F32
[3, 768]
a.blk.9.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.9.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.9.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.9.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.9.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.9.ln1.weight
F32
F32
[768]
a.blk.9.ln2.weight
F32
F32
[768]
a.blk.10
a.blk.10.conv_dw.weight
F32
F32
[3, 768]
a.blk.10.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.10.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.10.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.10.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.10.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.10.ln1.weight
F32
F32
[768]
a.blk.10.ln2.weight
F32
F32
[768]
a.blk.11
a.blk.11.attn_k.weight
BF16
BF16
[768, 384]
a.blk.11.attn_k_norm.weight
F32
F32
[96]
a.blk.11.attn_out.weight
BF16
BF16
[768, 768]
a.blk.11.attn_q.weight
BF16
BF16
[768, 768]
a.blk.11.attn_q_norm.weight
F32
F32
[96]
a.blk.11.attn_v.weight
BF16
BF16
[768, 384]
a.blk.11.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.11.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.11.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.11.ln1.weight
F32
F32
[768]
a.blk.11.ln2.weight
F32
F32
[768]
a.blk.12
a.blk.12.conv_dw.weight
F32
F32
[3, 768]
a.blk.12.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.12.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.12.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.12.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.12.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.12.ln1.weight
F32
F32
[768]
a.blk.12.ln2.weight
F32
F32
[768]
a.blk.13
a.blk.13.conv_dw.weight
F32
F32
[3, 768]
a.blk.13.conv_pw1.weight
BF16
BF16
[768, 2304]
a.blk.13.conv_pw2.weight
BF16
BF16
[768, 768]
a.blk.13.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.13.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.13.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.13.ln1.weight
F32
F32
[768]
a.blk.13.ln2.weight
F32
F32
[768]
a.blk.14
a.blk.14.attn_k.weight
BF16
BF16
[768, 384]
a.blk.14.attn_k_norm.weight
F32
F32
[96]
a.blk.14.attn_out.weight
BF16
BF16
[768, 768]
a.blk.14.attn_q.weight
BF16
BF16
[768, 768]
a.blk.14.attn_q_norm.weight
F32
F32
[96]
a.blk.14.attn_v.weight
BF16
BF16
[768, 384]
a.blk.14.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.14.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.14.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.14.ln1.weight
F32
F32
[768]
a.blk.14.ln2.weight
F32
F32
[768]
a.blk.15
a.blk.15.attn_k.weight
BF16
BF16
[768, 384]
a.blk.15.attn_k_norm.weight
F32
F32
[96]
a.blk.15.attn_out.weight
BF16
BF16
[768, 768]
a.blk.15.attn_q.weight
BF16
BF16
[768, 768]
a.blk.15.attn_q_norm.weight
F32
F32
[96]
a.blk.15.attn_v.weight
BF16
BF16
[768, 384]
a.blk.15.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.15.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.15.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.15.ln1.weight
F32
F32
[768]
a.blk.15.ln2.weight
F32
F32
[768]
a.blk.16
a.blk.16.attn_k.weight
BF16
BF16
[768, 384]
a.blk.16.attn_k_norm.weight
F32
F32
[96]
a.blk.16.attn_out.weight
BF16
BF16
[768, 768]
a.blk.16.attn_q.weight
BF16
BF16
[768, 768]
a.blk.16.attn_q_norm.weight
F32
F32
[96]
a.blk.16.attn_v.weight
BF16
BF16
[768, 384]
a.blk.16.ffn_down.weight
BF16
BF16
[1792, 768]
a.blk.16.ffn_gate.weight
BF16
BF16
[768, 1792]
a.blk.16.ffn_up.weight
BF16
BF16
[768, 1792]
a.blk.16.ln1.weight
F32
F32
[768]
a.blk.16.ln2.weight
F32
F32
[768]
a.conv1d.0.bias
F32
F32
[1, 1, 256]
a.conv1d.0.weight
F32
F32
[3, 3, 1, 256]
a.conv1d.2.bias
F32
F32
[1, 1, 256]
a.conv1d.2.weight
F32
F32
[3, 3, 1, 256]
a.conv1d.3.bias
F32
F32
[1, 1, 256]
a.conv1d.3.weight
F32
F32
[1, 1, 256, 256]
a.conv1d.5.bias
F32
F32
[1, 1, 256]
a.conv1d.5.weight
F32
F32
[3, 3, 1, 256]
a.conv1d.6.bias
F32
F32
[1, 1, 256]
a.conv1d.6.weight
F32
F32
[1, 1, 256, 256]
a.gen.code.blk.0
a.gen.code.blk.0.attn_gate.weight
BF16
BF16
[1024, 32]
a.gen.code.blk.0.attn_k.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.0.attn_k_norm.weight
F32
F32
[32]
a.gen.code.blk.0.attn_out.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.0.attn_q.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.0.attn_q_norm.weight
F32
F32
[32]
a.gen.code.blk.0.attn_v.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.0.ffn_down.weight
BF16
BF16
[2816, 1024]
a.gen.code.blk.0.ffn_gate.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.0.ffn_up.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.0.ln1.weight
F32
F32
[1024]
a.gen.code.blk.0.ln2.weight
F32
F32
[1024]
a.gen.code.blk.1
a.gen.code.blk.1.attn_gate.weight
BF16
BF16
[1024, 32]
a.gen.code.blk.1.attn_k.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.1.attn_k_norm.weight
F32
F32
[32]
a.gen.code.blk.1.attn_out.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.1.attn_q.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.1.attn_q_norm.weight
F32
F32
[32]
a.gen.code.blk.1.attn_v.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.1.ffn_down.weight
BF16
BF16
[2816, 1024]
a.gen.code.blk.1.ffn_gate.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.1.ffn_up.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.1.ln1.weight
F32
F32
[1024]
a.gen.code.blk.1.ln2.weight
F32
F32
[1024]
a.gen.code.blk.2
a.gen.code.blk.2.attn_gate.weight
BF16
BF16
[1024, 32]
a.gen.code.blk.2.attn_k.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.2.attn_k_norm.weight
F32
F32
[32]
a.gen.code.blk.2.attn_out.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.2.attn_q.weight
BF16
BF16
[1024, 1024]
a.gen.code.blk.2.attn_q_norm.weight
F32
F32
[32]
a.gen.code.blk.2.attn_v.weight
BF16
BF16
[1024, 256]
a.gen.code.blk.2.ffn_down.weight
BF16
BF16
[2816, 1024]
a.gen.code.blk.2.ffn_gate.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.2.ffn_up.weight
BF16
BF16
[1024, 2816]
a.gen.code.blk.2.ln1.weight
F32
F32
[1024]
a.gen.code.blk.2.ln2.weight
F32
F32
[1024]
a.gen.code.embd.weight
BF16
BF16
[1024, 2017, 12]
a.gen.code.head_norm.weight
F32
F32
[1024, 12]
a.gen.code.mask_embd
F32
F32
[1024]
a.gen.code.out_embd.weight
BF16
BF16
[2048, 24204]
a.gen.code.output_norm.weight
F32
F32
[1024]
a.gen.code.proj_in.bias
F32
F32
[12288]
a.gen.code.proj_in.weight
BF16
BF16
[2048, 12288]
a.gen.wav.quant.first.codebook.weight
F32
F32
[4, 2016]
a.gen.wav.quant.first.out_proj.bias
F32
F32
[256]
a.gen.wav.quant.first.out_proj.weight
F32
F32
[48, 256]
a.gen.wav.tfm.blk.0
a.gen.wav.tfm.blk.0.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.0.conv_dw.weight
F32
F32
[7, 512]
a.gen.wav.tfm.blk.0.conv_pw1.weight
F16
F16
[512, 1536]
a.gen.wav.tfm.blk.0.conv_pw2.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.0.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.0.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.0.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.0.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.0.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.0.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.1
a.gen.wav.tfm.blk.1.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.1.conv_dw.weight
F32
F32
[7, 512]
a.gen.wav.tfm.blk.1.conv_pw1.weight
F16
F16
[512, 1536]
a.gen.wav.tfm.blk.1.conv_pw2.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.1.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.1.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.1.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.1.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.1.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.1.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.2
a.gen.wav.tfm.blk.2.attn_gate.weight
F16
F16
[512, 8]
a.gen.wav.tfm.blk.2.attn_k.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.2.attn_k_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.2.attn_out.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.2.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.2.attn_q.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.2.attn_q_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.2.attn_v.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.2.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.2.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.2.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.2.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.2.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.2.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.3
a.gen.wav.tfm.blk.3.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.3.conv_dw.weight
F32
F32
[7, 512]
a.gen.wav.tfm.blk.3.conv_pw1.weight
F16
F16
[512, 1536]
a.gen.wav.tfm.blk.3.conv_pw2.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.3.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.3.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.3.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.3.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.3.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.3.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.4
a.gen.wav.tfm.blk.4.attn_gate.weight
F16
F16
[512, 8]
a.gen.wav.tfm.blk.4.attn_k.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.4.attn_k_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.4.attn_out.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.4.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.4.attn_q.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.4.attn_q_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.4.attn_v.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.4.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.4.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.4.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.4.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.4.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.4.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.5
a.gen.wav.tfm.blk.5.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.5.conv_dw.weight
F32
F32
[7, 512]
a.gen.wav.tfm.blk.5.conv_pw1.weight
F16
F16
[512, 1536]
a.gen.wav.tfm.blk.5.conv_pw2.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.5.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.5.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.5.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.5.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.5.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.5.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.6
a.gen.wav.tfm.blk.6.attn_gate.weight
F16
F16
[512, 8]
a.gen.wav.tfm.blk.6.attn_k.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.6.attn_k_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.6.attn_out.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.6.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.6.attn_q.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.6.attn_q_norm.weight
F32
F32
[64]
a.gen.wav.tfm.blk.6.attn_v.weight
F16
F16
[512, 128]
a.gen.wav.tfm.blk.6.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.6.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.6.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.6.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.6.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.6.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.blk.7
a.gen.wav.tfm.blk.7.attn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.7.conv_dw.weight
F32
F32
[7, 512]
a.gen.wav.tfm.blk.7.conv_pw1.weight
F16
F16
[512, 1536]
a.gen.wav.tfm.blk.7.conv_pw2.weight
F16
F16
[512, 512]
a.gen.wav.tfm.blk.7.ffn_down.weight
F16
F16
[2304, 512]
a.gen.wav.tfm.blk.7.ffn_gate.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.7.ffn_post_norm.weight
F32
F32
[512]
a.gen.wav.tfm.blk.7.ffn_up.weight
F16
F16
[512, 2304]
a.gen.wav.tfm.blk.7.ln1.weight
F32
F32
[512]
a.gen.wav.tfm.blk.7.ln2.weight
F32
F32
[512]
a.gen.wav.tfm.in_proj.bias
F32
F32
[512]
a.gen.wav.tfm.in_proj.weight
F16
F16
[256, 512]
a.gen.wav.tfm.out_proj.bias
F32
F32
[1282]
a.gen.wav.tfm.out_proj.weight
F16
F16
[512, 1282]
a.gen.wav.tfm.output_norm.weight
F32
F32
[512]
a.post_ln.weight
F32
F32
[768]
a.pre_encode.out.bias
F32
F32
[768]
a.pre_encode.out.weight
BF16
BF16
[4096, 768]
mm.0.bias
F32
F32
[4096]
mm.0.weight
BF16
BF16
[4096, 4096]
mm.2.bias
F32
F32
[2048]
mm.2.weight
BF16
BF16
[4096, 2048]
mm.a.mlp.0.bias
F32
F32
[768]
mm.a.mlp.0.weight
F32
F32
[768]
mm.a.mlp.1.bias
F32
F32
[2048]
mm.a.mlp.1.weight
BF16
BF16
[768, 2048]
mm.a.mlp.3.bias
F32
F32
[2048]
mm.a.mlp.3.weight
BF16
BF16
[2048, 2048]
mm.input_norm.bias
F32
F32
[1024]
mm.input_norm.weight
F32
F32
[1024]
v.blk.0
v.blk.0.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.0.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.0.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.0.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.0.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.0.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.0.ln1.weight
F32
F32
[1024]
v.blk.0.ln2.weight
F32
F32
[1024]
v.blk.1
v.blk.1.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.1.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.1.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.1.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.1.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.1.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.1.ln1.weight
F32
F32
[1024]
v.blk.1.ln2.weight
F32
F32
[1024]
v.blk.2
v.blk.2.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.2.attn_k_norm.weight
F32
F32
[64]
v.blk.2.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.2.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.2.attn_q_norm.weight
F32
F32
[64]
v.blk.2.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.2.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.2.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.2.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.2.ln1.weight
F32
F32
[1024]
v.blk.2.ln2.weight
F32
F32
[1024]
v.blk.3
v.blk.3.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.3.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.3.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.3.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.3.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.3.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.3.ln1.weight
F32
F32
[1024]
v.blk.3.ln2.weight
F32
F32
[1024]
v.blk.4
v.blk.4.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.4.attn_k_norm.weight
F32
F32
[64]
v.blk.4.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.4.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.4.attn_q_norm.weight
F32
F32
[64]
v.blk.4.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.4.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.4.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.4.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.4.ln1.weight
F32
F32
[1024]
v.blk.4.ln2.weight
F32
F32
[1024]
v.blk.5
v.blk.5.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.5.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.5.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.5.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.5.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.5.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.5.ln1.weight
F32
F32
[1024]
v.blk.5.ln2.weight
F32
F32
[1024]
v.blk.6
v.blk.6.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.6.attn_k_norm.weight
F32
F32
[64]
v.blk.6.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.6.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.6.attn_q_norm.weight
F32
F32
[64]
v.blk.6.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.6.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.6.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.6.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.6.ln1.weight
F32
F32
[1024]
v.blk.6.ln2.weight
F32
F32
[1024]
v.blk.7
v.blk.7.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.7.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.7.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.7.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.7.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.7.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.7.ln1.weight
F32
F32
[1024]
v.blk.7.ln2.weight
F32
F32
[1024]
v.blk.8
v.blk.8.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.8.attn_k_norm.weight
F32
F32
[64]
v.blk.8.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.8.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.8.attn_q_norm.weight
F32
F32
[64]
v.blk.8.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.8.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.8.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.8.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.8.ln1.weight
F32
F32
[1024]
v.blk.8.ln2.weight
F32
F32
[1024]
v.blk.9
v.blk.9.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.9.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.9.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.9.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.9.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.9.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.9.ln1.weight
F32
F32
[1024]
v.blk.9.ln2.weight
F32
F32
[1024]
v.blk.10
v.blk.10.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.10.attn_k_norm.weight
F32
F32
[64]
v.blk.10.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.10.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.10.attn_q_norm.weight
F32
F32
[64]
v.blk.10.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.10.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.10.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.10.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.10.ln1.weight
F32
F32
[1024]
v.blk.10.ln2.weight
F32
F32
[1024]
v.blk.11
v.blk.11.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.11.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.11.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.11.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.11.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.11.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.11.ln1.weight
F32
F32
[1024]
v.blk.11.ln2.weight
F32
F32
[1024]
v.blk.12
v.blk.12.attn_k.weight
BF16
BF16
[1024, 512]
v.blk.12.attn_k_norm.weight
F32
F32
[64]
v.blk.12.attn_out.weight
BF16
BF16
[1024, 1024]
v.blk.12.attn_q.weight
BF16
BF16
[1024, 1024]
v.blk.12.attn_q_norm.weight
F32
F32
[64]
v.blk.12.attn_v.weight
BF16
BF16
[1024, 512]
v.blk.12.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.12.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.12.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.12.ln1.weight
F32
F32
[1024]
v.blk.12.ln2.weight
F32
F32
[1024]
v.blk.13
v.blk.13.conv2d_in_proj.weight
BF16
BF16
[1024, 3072]
v.blk.13.conv2d_out_proj.weight
BF16
BF16
[1024, 1024]
v.blk.13.conv2d_spatial.weight
F32
F32
[3, 3, 1, 1024]
v.blk.13.ffn_down.weight
BF16
BF16
[2560, 1024]
v.blk.13.ffn_gate.weight
BF16
BF16
[1024, 2560]
v.blk.13.ffn_up.weight
BF16
BF16
[1024, 2560]
v.blk.13.ln1.weight
F32
F32
[1024]
v.blk.13.ln2.weight
F32
F32
[1024]
v.embedding_norm.weight
F32
F32
[1024]
v.patch_embd.weight
F16
F16
[16, 16, 3, 1024]
v.patch_embd.weight.1
F16
F16
[16, 16, 3, 1024]