From 1ca3f06a54afcd7135ff63e5c4263754f5c40235 Mon Sep 17 00:00:00 2001 From: caitianchi Date: Tue, 13 Aug 2024 09:52:52 +0800 Subject: [PATCH] fix type-check --- .../llava/minicpmv-convert-image-encoder-to-gguf.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/examples/llava/minicpmv-convert-image-encoder-to-gguf.py b/examples/llava/minicpmv-convert-image-encoder-to-gguf.py index 8d7e7204f..ea773742a 100644 --- a/examples/llava/minicpmv-convert-image-encoder-to-gguf.py +++ b/examples/llava/minicpmv-convert-image-encoder-to-gguf.py @@ -551,7 +551,7 @@ elif minicpmv_version == 2: emb_dim = 4096 elif minicpmv_version == 3: emb_dim = 3584 - + default_vision_config = { "hidden_size": 1152, "image_size": 980, @@ -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