Skip to content

Commit 8031aff

Browse files
committed
add minicpmv back
1 parent 56fc382 commit 8031aff

File tree

2 files changed

+17
-20
lines changed

2 files changed

+17
-20
lines changed

tests/openvino/test_modeling.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -2130,9 +2130,7 @@ class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase):
21302130
if is_transformers_version(">=", "4.40.0"):
21312131
SUPPORTED_ARCHITECTURES += ["llava_next", "nanollava"]
21322132
if is_transformers_version(">=", "4.45.0"):
2133-
SUPPORTED_ARCHITECTURES += ["internvl2", "phi3_v", "qwen2_vl"]
2134-
# TODO: add back when https://huggingface.co/katuni4ka/tiny-random-minicpmv-2_6/discussions/1 merged
2135-
# SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl"]
2133+
SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl"]
21362134

21372135
if is_transformers_version(">=", "4.46.0"):
21382136
SUPPORTED_ARCHITECTURES += ["maira2"]

tests/openvino/test_quantization.py

+16-17
Original file line numberDiff line numberDiff line change
@@ -583,23 +583,22 @@ class OVWeightCompressionTest(unittest.TestCase):
583583
if is_transformers_version(">=", "4.45.0"):
584584
LOAD_IN_4_BITS_SCOPE.extend(
585585
[
586-
# TODO: add back when https://huggingface.co/katuni4ka/tiny-random-minicpmv-2_6/discussions/1 merged
587-
# (
588-
# OVModelForVisualCausalLM,
589-
# "minicpmv",
590-
# True,
591-
# dict(
592-
# bits=4,
593-
# group_size=16,
594-
# dataset="contextual",
595-
# ratio=0.8,
596-
# sensitivity_metric="mean_activation_magnitude",
597-
# num_samples=1,
598-
# processor=MODEL_NAMES["minicpmv"],
599-
# trust_remote_code=True,
600-
# ),
601-
# [{"int8": 8, "int4": 22}, {"int8": 1}, {"int8": 26}, {"int8": 6}],
602-
# ),
586+
(
587+
OVModelForVisualCausalLM,
588+
"minicpmv",
589+
True,
590+
dict(
591+
bits=4,
592+
group_size=16,
593+
dataset="contextual",
594+
ratio=0.8,
595+
sensitivity_metric="mean_activation_magnitude",
596+
num_samples=1,
597+
processor=MODEL_NAMES["minicpmv"],
598+
trust_remote_code=True,
599+
),
600+
[{"int8": 8, "int4": 22}, {"int8": 1}, {"int8": 26}, {"int8": 6}],
601+
),
603602
(
604603
OVModelForVisualCausalLM,
605604
"internvl2",

0 commit comments

Comments
 (0)