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

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

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