pub struct DflashWeights {Show 13 fields
pub config: DflashConfig,
pub fc: DenseWeight,
pub hidden_norm: DenseWeight,
pub norm: DenseWeight,
pub layers: Vec<DflashLayerWeights>,
pub draft_id_to_target_id: Option<Vec<i64>>,
pub markov_w1: Option<DenseWeight>,
pub markov_w2: Option<DenseWeight>,
pub confidence_proj: Option<DenseWeight>,
pub confidence_bias: Option<DenseWeight>,
pub selector_pred: Option<DenseWeight>,
pub selector_succ: Option<DenseWeight>,
pub selector_hidden_proj: Option<DenseWeight>,
}Expand description
2026-09-25: The drafter weights load_dflash_weights returns. It holds no
embedding or LM head.
Fields§
§config: DflashConfig§fc: DenseWeight§norm: DenseWeight§layers: Vec<DflashLayerWeights>§draft_id_to_target_id: Option<Vec<i64>>2026-09-25: Some (and empty) when the store has a d2t or
draft_id_to_target_id tensor; the table itself is not loaded.
markov_w1: Option<DenseWeight>2026-09-25: markov_head.markov_w1.weight. markov_w1 and markov_w2 are
Some when markov_rank > 0 and the store has markov_w1.
markov_w2: Option<DenseWeight>2026-09-25: markov_head.markov_w2.weight.
confidence_proj: Option<DenseWeight>2026-09-25: confidence_head.proj.weight. The weight and the bias are
Some when enable_confidence_head is set and the store has the weight.
confidence_bias: Option<DenseWeight>2026-09-25: confidence_head.proj.bias.
selector_pred: Option<DenseWeight>2026-09-25: candidate_selector.predecessor_codebook. The three selector
tensors are Some when dflash_config sets selector_rank and
selector_top_k and the store has this codebook.
selector_succ: Option<DenseWeight>2026-09-25: candidate_selector.successor_codebook.
2026-09-25: candidate_selector.hidden_projection.weight.