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

fix type-check

Browse source
This commit is contained in:
caitianchi 2024-08-13 09:52:52 +08:00
parent 47eb0a5530
commit 1ca3f06a54
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -551,7 +551,7 @@ elif minicpmv_version == 2:
emb_dim = 4096 emb_dim = 4096
elif minicpmv_version == 3: elif minicpmv_version == 3:
emb_dim = 3584 emb_dim = 3584
default_vision_config = { default_vision_config = {
"hidden_size": 1152, "hidden_size": 1152,
"image_size": 980, "image_size": 980,
@ -562,10 +562,9 @@ default_vision_config = {
"patch_size": 14, "patch_size": 14,
} }
if minicpmv_version == 2: vision_config = Idefics2VisionConfig(**default_vision_config)
vision_config = Idefics2VisionConfig(**default_vision_config) model = Idefics2VisionTransformer(vision_config)
model = Idefics2VisionTransformer(vision_config) if minicpmv_version == 3:
elif minicpmv_version == 3:
vision_config = SiglipVisionConfig(**default_vision_config) vision_config = SiglipVisionConfig(**default_vision_config)
model = SiglipVisionTransformer(vision_config) model = SiglipVisionTransformer(vision_config)
@ -581,7 +580,6 @@ has_text_encoder = True
has_vision_encoder = True has_vision_encoder = True
has_minicpmv_projector = False has_minicpmv_projector = False
if args.text_only: if args.text_only:
fname_middle = "text-" fname_middle = "text-"
has_vision_encoder = False has_vision_encoder = False