fix Type-Check error

OpenBMB · Aug 7, 2024 · 28230d0 · 28230d0
1 parent 5ab9577
commit 28230d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 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 @@ def _replace_name_resampler(s, v):
         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()