fix type-check

This commit is contained in:
caitianchi 2024-08-10 21:51:04 +08:00
parent 4a87d1d93e
commit 32b47f600f
5 changed files with 9 additions and 10 deletions

View file

@ -776,7 +776,6 @@ static ggml_cgraph * clip_image_build_graph(clip_ctx * ctx, const clip_image_f32
embeddings = ggml_gelu(ctx0, embeddings);
embeddings = ggml_mul_mat(ctx0, model.mm_2_w, embeddings);
embeddings = ggml_add(ctx0, embeddings, model.mm_2_b);
}
else if (ctx->proj_type == PROJECTOR_TYPE_MLP_NORM) {
embeddings = ggml_mul_mat(ctx0, model.mm_0_w, embeddings);

View file

@ -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"
@ -159,6 +158,7 @@ fname_middle = None
has_text_encoder = True
has_vision_encoder = True
has_minicpmv_projector = False
minicpmv_version = 2
if args.text_only:
fname_middle = "text-"
has_vision_encoder = False

View file

@ -134,7 +134,6 @@ class SiglipVisionConfig(PretrainedConfig):
return cls.from_dict(config_dict, **kwargs)
_CHECKPOINT_FOR_DOC = "google/siglip-base-patch16-224"
SIGLIP_PRETRAINED_MODEL_ARCHIVE_LIST = [
@ -1096,6 +1095,7 @@ fname_middle = None
has_text_encoder = True
has_vision_encoder = True
has_minicpmv_projector = False
minicpmv_version = 3
if args.text_only:
fname_middle = "text-"
has_vision_encoder = False