Skip to content

Commit 7098e94

Browse files
[OpenVINO] Update SAM low precision references (#1499)
* Update SAM low precision references * Fix * Fix 2
1 parent 72e08c4 commit 7098e94

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

tests/openvino/test_exporters_cli.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
from optimum.intel.openvino.utils import _HEAD_TO_AUTOMODELS, TemporaryDirectory
6464
from optimum.intel.utils.import_utils import (
6565
compare_versions,
66+
is_nncf_version,
6667
is_openvino_tokenizers_available,
6768
is_openvino_version,
6869
is_tokenizers_version,
@@ -433,11 +434,11 @@ class OVCLIExportTestCase(unittest.TestCase):
433434
"--dataset coco --num-samples 1",
434435
{
435436
"vision_encoder": 75,
436-
"prompt_encoder_mask_decoder": 61,
437+
"prompt_encoder_mask_decoder": 61 if is_nncf_version("<=", "2.18") else 60,
437438
},
438439
{
439440
"vision_encoder": {"int8": 75},
440-
"prompt_encoder_mask_decoder": {"int8": 50},
441+
"prompt_encoder_mask_decoder": {"int8": 50 if is_nncf_version("<=", "2.18") else 49},
441442
},
442443
),
443444
(

tests/openvino/test_quantization.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@
8585
from copy import deepcopy
8686

8787
from optimum.intel.openvino.quantization import InferRequestWrapper, OVCalibrationDatasetBuilder
88-
from optimum.intel.utils.import_utils import is_openvino_version, is_transformers_version
88+
from optimum.intel.utils.import_utils import is_openvino_version, is_transformers_version, is_nncf_version
8989
from utils_tests import (
9090
MODEL_NAMES,
9191
get_num_quantized_nodes,
@@ -364,11 +364,11 @@ class OVQuantizerTest(unittest.TestCase):
364364
OVQuantizationConfig(bits=8, dataset="coco", num_samples=1),
365365
{
366366
"vision_encoder": 75,
367-
"prompt_encoder_mask_decoder": 61,
367+
"prompt_encoder_mask_decoder": 61 if is_nncf_version("<=", "2.18") else 60,
368368
},
369369
{
370370
"vision_encoder": {"int8": 75},
371-
"prompt_encoder_mask_decoder": {"int8": 50},
371+
"prompt_encoder_mask_decoder": {"int8": 50 if is_nncf_version("<=", "2.18") else 49},
372372
},
373373
),
374374
(
@@ -737,7 +737,7 @@ class OVWeightCompressionTest(unittest.TestCase):
737737
dict(bits=4, dataset="coco", num_samples=1, group_size=2),
738738
{
739739
"vision_encoder": {"int8": 56, "int4": 94},
740-
"prompt_encoder_mask_decoder": {"int8": 6, "int4": 94},
740+
"prompt_encoder_mask_decoder": {"int8": 6, "int4": 94 if is_nncf_version("<=", "2.18") else 92},
741741
},
742742
),
743743
(

tests/openvino/utils_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@
319319
},
320320
"sam": {
321321
"vision_encoder": 102 if is_openvino_version("<", "2025.2.0") else 150,
322-
"prompt_encoder_mask_decoder": 100,
322+
"prompt_encoder_mask_decoder": 100 if is_nncf_version("<=", "2.18") else 98,
323323
},
324324
"speecht5": {
325325
"encoder": 28,

0 commit comments

Comments
 (0)