Revert changes to KimiVLForConditionalGeneration
This commit is contained in:
parent
9c44981c01
commit
9b14cb8b28
|
|
@ -10705,17 +10705,9 @@ class KimiVLModel(MmprojModel):
|
|||
self.gguf_writer.add_vision_attention_layernorm_eps(self.hparams_vision.get("layer_norm_eps", 1e-5))
|
||||
|
||||
def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]:
|
||||
is_vision_tensor = "vision_tower" in name or "multi_modal_projector" in name or "mm_projector" in name
|
||||
is_vision_tensor = "vision_tower" in name or "multi_modal_projector" in name
|
||||
|
||||
if is_vision_tensor:
|
||||
# update names:
|
||||
# "mm_projector.proj.0" -> "mm_projector.proj.linear_1.",
|
||||
# "mm_projector.proj.2" -> "mm_projector.proj.linear_2.",
|
||||
if "proj.0." in name:
|
||||
name = name.replace(".0.", ".linear_1.")
|
||||
if "proj.2." in name:
|
||||
name = name.replace(".2.", ".linear_2.")
|
||||
|
||||
if "pos_emb.weight" in name:
|
||||
data_torch = data_torch.view(data_torch.shape[0] * data_torch.shape[1], data_torch.shape[2])
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue