Skip to content

Commit 07a7412

Browse files
authored
mtmd: add min/max pixels gguf metadata (ggml-org#19273)
1 parent 9f682fb commit 07a7412

3 files changed

Lines changed: 10 additions & 0 deletions

File tree

gguf-py/gguf/constants.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,8 @@ class Clip:
284284
class ClipVision:
285285
PROJECTOR_TYPE = "clip.vision.projector_type" # for mixed modality models
286286
IMAGE_SIZE = "clip.vision.image_size"
287+
IMAGE_MIN_PIXELS = "clip.vision.image_min_pixels"
288+
IMAGE_MAX_PIXELS = "clip.vision.image_max_pixels"
287289
PREPROC_IMAGE_SIZE = "clip.vision.preproc_image_size"
288290
PATCH_SIZE = "clip.vision.patch_size"
289291
EMBEDDING_LENGTH = "clip.vision.embedding_length"

gguf-py/gguf/gguf_writer.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1113,6 +1113,12 @@ def add_vision_attention_layernorm_eps(self, value: float) -> None:
11131113
def add_vision_image_size(self, value: int) -> None:
11141114
self.add_uint32(Keys.ClipVision.IMAGE_SIZE, value)
11151115

1116+
def add_vision_max_pixels(self, value: int) -> None:
1117+
self.add_uint32(Keys.ClipVision.IMAGE_MAX_PIXELS, value)
1118+
1119+
def add_vision_min_pixels(self, value: int) -> None:
1120+
self.add_uint32(Keys.ClipVision.IMAGE_MIN_PIXELS, value)
1121+
11161122
def add_vision_preproc_image_size(self, value: int) -> None:
11171123
self.add_uint32(Keys.ClipVision.PREPROC_IMAGE_SIZE, value)
11181124

tools/mtmd/clip-impl.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
// vision-specific
3737
#define KEY_VISION_PROJ_TYPE "clip.vision.projector_type" // for models with mixed modalities
3838
#define KEY_IMAGE_SIZE "clip.vision.image_size"
39+
#define KEY_IMAGE_MIN_PIXELS "clip.vision.image_min_pixels"
40+
#define KEY_IMAGE_MAX_PIXELS "clip.vision.image_max_pixels"
3941
#define KEY_PREPROC_IMAGE_SIZE "clip.vision.preproc_image_size"
4042
#define KEY_PATCH_SIZE "clip.vision.patch_size"
4143
#define KEY_IMAGE_MEAN "clip.vision.image_mean"

0 commit comments

Comments
 (0)