← MIC coverage of Transformers model types

google/umt5-small

huggingface-mic/umt5-small-mic@3ea88ecd4cac ← google/umt5-small@8c63c2b77efb · UMT5ForConditionalGeneration · random weights · built before the shrink rule (no recorded recipe)

CI: no failure on main in 30 days · MIC failures dashboard

Original175.3Mparameters
Fixture16.6Mparameters
Layer stacks2compared
Findings0to review

✓ Same architecture, smaller

Every module class, layer kind and checked property of the original is in the fixture.

Layer stacks

One square per layer, colored by kind: the layer's module classes and type attributes. Hover a square for its index and kind.

encoder.block

Original · 8
Fixture · 2
KindOriginalFixture
Block82✓ kept

decoder.block

Original · 8
Fixture · 2
KindOriginalFixture
Block82✓ kept

Where the parameters are

Share of each model's parameters per component. Vocabulary and patch sizes are kept, so embeddings usually take a larger share of the fixture.

Original
Fixture
ComponentOriginalShareFixtureShare
shared131.3M74.9%16.4M98.8%
decoder.block25.2M14.4%115.3K0.7%
encoder.block18.9M10.8%82.4K0.5%
decoder.final_layer_norm5120.0%640.0%
encoder.final_layer_norm5120.0%640.0%

Preserved properties

ConfigPropertyOriginalFixture
modelmodel_typeumt5umt5✓ kept
modelvocab_size256384256384✓ kept
modeltie_word_embeddingsTrueTrue✓ kept

Module classes

Instances of each class in the built model (prefix UMT5 omitted).

ClassOriginalFixture
Linear14537
Dropout5816
LayerNorm4212
Attention246
Embedding197
ModuleList186
NewGELUActivation164
Block164
DenseGatedActDense164
LayerFF164
LayerSelfAttention164
LayerCrossAttention82
Stack22
ForConditionalGeneration11

Config changes

6 fields changed, 28 unchanged.

FieldOriginalFixture
d_ff1024128
d_kv6416
d_model51264
num_decoder_layers82
num_heads64
num_layers82