From 28230d0b1351993928113653f285359b1b2bdab2 Mon Sep 17 00:00:00 2001 From: caitianchi Date: Wed, 7 Aug 2024 20:07:27 +0800 Subject: [PATCH] fix Type-Check error --- examples/llava/minicpmv-convert-image-encoder-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/llava/minicpmv-convert-image-encoder-to-gguf.py b/examples/llava/minicpmv-convert-image-encoder-to-gguf.py index 71af0b4f1..abb35c6ca 100644 --- a/examples/llava/minicpmv-convert-image-encoder-to-gguf.py +++ b/examples/llava/minicpmv-convert-image-encoder-to-gguf.py @@ -6,7 +6,6 @@ import re import torch import numpy as np from gguf import * -import timm from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer, Idefics2VisionConfig TEXT = "clip.text" @@ -298,6 +297,7 @@ if has_minicpmv_projector: for nk, nv in kvs.items(): new_state_dict[nk] = nv projector = new_state_dict + ftype_cur = 0 for name, data in projector.items(): name = get_tensor_name(name) data = data.squeeze().numpy()