pub fn extract_encoder_weights(
weights: &mut WeightMap,
cfg: &Vjepa2Config,
) -> Result<Vjepa2EncoderWeights, Error>pub fn extract_encoder_weights(
weights: &mut WeightMap,
cfg: &Vjepa2Config,
) -> Result<Vjepa2EncoderWeights, Error>