diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 1e10af5f9f..dcd6dc8b26 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -1135,7 +1135,7 @@ class GGUFWriter: def add_vision_sam_embedding_length(self, value: int) -> None: self.add_uint32(Keys.ClipVision.SAM.EMBEDDING_LENGTH, value) - + def add_vision_sam_head_count(self, value: int) -> None: self.add_uint32(Keys.ClipVision.SAM.HEAD_COUNT, value) diff --git a/tools/mtmd/clip-impl.h b/tools/mtmd/clip-impl.h index a90e67f360..f8c2a105df 100644 --- a/tools/mtmd/clip-impl.h +++ b/tools/mtmd/clip-impl.h @@ -51,7 +51,7 @@ #define KEY_MINICPMV_QUERY_NUM "clip.minicpmv_query_num" #define KEY_SAM_N_HEAD "clip.vision.sam.head_count" #define KEY_SAM_N_BLOCK "clip.vision.sam.block_count" -#define KEY_SAM_N_EMBD "clip.vision.sam.embedding_length" +#define KEY_SAM_N_EMBD "clip.vision.sam.embedding_length" // audio-specific #define KEY_AUDIO_PROJ_TYPE "clip.audio.projector_type" // for models with mixed modalities #define KEY_A_NUM_MEL_BINS "clip.audio.num_mel_bins"