vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix Type-Check error

Browse source
This commit is contained in:
caitianchi 2024-08-07 20:07:27 +08:00
parent 5ab95776f7
commit 28230d0b13
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/llava/minicpmv-convert-image-encoder-to-gguf.py
View file

@ -6,7 +6,6 @@ import re
import torch import torch
import numpy as np import numpy as np
from gguf import * from gguf import *
import timm
from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer, Idefics2VisionConfig from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer, Idefics2VisionConfig
TEXT = "clip.text" TEXT = "clip.text"
@ -298,6 +297,7 @@ if has_minicpmv_projector:
for nk, nv in kvs.items(): for nk, nv in kvs.items():
new_state_dict[nk] = nv new_state_dict[nk] = nv
projector = new_state_dict projector = new_state_dict
ftype_cur = 0
for name, data in projector.items(): for name, data in projector.items():
name = get_tensor_name(name) name = get_tensor_name(name)
data = data.squeeze().numpy() data = data.squeeze().numpy()