Kimi-K2.5-2layer-BF16

A 2-layer BF16 slice of moonshotai/Kimi-K2.5, built as a GPU CI fixture for SkyRL. This is not a usable language model β€” two layers of a 61-layer network produce incoherent text. It exists so tests can exercise the real architecture, with real (truncated) weights, at a size that fits a single node.

Contents

  • language_model.model.layers.0 β€” dense MLP layer (first_k_dense_replace=1)
  • language_model.model.layers.1 β€” MoE layer with all 384 routed experts + 1 shared expert
  • language_model.model.embed_tokens, language_model.lm_head, language_model.model.norm
  • vision_tower.* and mm_projector.*, unchanged

text_config.num_hidden_layers is set to 2; everything else in the config is untouched.

BF16 conversion

Only the routed-expert weights are quantized in the source release (its ignore list covers attention, shared experts, dense MLP, lm_head and the vision tower). Those tensors were dequantized from compressed-tensors pack-quantized INT4 with the same arithmetic SkyRL uses (examples/train/megatron/dequantize_compressed_tensors_int4.py: nibble = q + 8, per-group q * weight_scale, group_size=32), and quantization_config was dropped from the config.

Every dequantized tensor was verified to be a fixed point of the fake-quant STE (scale_divisor=7.0, q_min=-7): re-quantizing reproduces the stored scales and codes bit-for-bit.

Total: 40.8 GB, ~20B parameters, 33 GB of which are the routed experts.

Downloads last month
956
Safetensors
Model size
20B params
Tensor type
F32
Β·
BF16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for eatang/Kimi-K2.5-2layer-BF16

Finetuned
(47)
this model