Skip to content

Commit 74940a0

Browse files
Update test references for NNCF 3.1.0
1 parent 1802051 commit 74940a0

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
"optimum-onnx@git+https://github.com/huggingface/optimum-onnx.git@main",
3232
"transformers>=4.45,<4.58",
3333
"setuptools",
34-
"nncf>=2.19.0",
34+
"nncf>=3.1.0",
3535
"openvino>=2025.4.0",
3636
"openvino-tokenizers>=2025.4.0",
3737
]
@@ -68,8 +68,8 @@
6868
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
6969

7070
EXTRAS_REQUIRE = {
71-
"nncf": ["nncf>=2.19.0"],
72-
"openvino": ["nncf>=2.19.0", "openvino>=2025.4.0", "openvino-tokenizers>=2025.4.0"],
71+
"nncf": ["nncf>=3.1.0"],
72+
"openvino": ["nncf>=3.1.0", "openvino>=2025.4.0", "openvino-tokenizers>=2025.4.0"],
7373
"neural-compressor": ["neural-compressor[pt]>=3.4.1,<3.8", "accelerate", "transformers<4.46", "datasets"],
7474
"ipex": ["intel-extension-for-pytorch>=2.8", "transformers>4.54,<4.56", "accelerate"],
7575
"diffusers": ["diffusers"],

tests/openvino/test_exporters_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,7 @@ class OVCLIExportTestCase(unittest.TestCase):
539539
"text-generation-with-past",
540540
"opt125m",
541541
"mxfp4",
542-
{"model": {"int8": 4, "f4e2m1": 72, "f8e8m0": 72}},
542+
{"model": {"int8": 0, "f4e2m1": 72, "f8e4m3": 2, "f8e8m0": 74}},
543543
),
544544
(
545545
"text-generation-with-past",

tests/openvino/test_quantization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -613,7 +613,7 @@ class OVWeightCompressionTest(unittest.TestCase):
613613
"gpt2",
614614
False,
615615
dict(bits=4, dtype="mxfp4", group_size=32),
616-
{"model": {"int8": 4, "f4e2m1": 20, "f8e8m0": 20}},
616+
{"model": {"int8": 0, "f4e2m1": 20, "f8e8m0": 22, "f8e4m3": 2}},
617617
),
618618
(
619619
OVModelForCausalLM,

0 commit comments

Comments
 (0)