convert : correct indentation

This commit is contained in:
Stanisław Szymczyk 2025-01-02 21:13:35 +01:00
parent 93aca64520
commit d2f784d50d

View file

@ -3879,7 +3879,7 @@ class DeepseekV2Model(Model):
def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]: def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]:
# rename e_score_correction_bias tensors # rename e_score_correction_bias tensors
if name.endswith("e_score_correction_bias"): if name.endswith("e_score_correction_bias"):
name = name.replace("e_score_correction_bias", "e_score_correction.bias") name = name.replace("e_score_correction_bias", "e_score_correction.bias")
# skip Multi-Token Prediction (MTP) layers # skip Multi-Token Prediction (MTP) layers
block_count = self.hparams["num_hidden_layers"] block_count = self.hparams["num_hidden_layers"]