Skip to content

Commit b1d1cc4

Browse files
Added NNCF to export-requirements.txt (openvinotoolkit#1974)
1 parent 31d4655 commit b1d1cc4

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

samples/export-requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@
22
--extra-index-url https://storage.openvinotoolkit.org/simple/wheels/pre-release
33
--extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
44
openvino-tokenizers~=2025.2.0.0.dev
5-
optimum-intel @ git+https://github.com/huggingface/optimum-intel.git@main
5+
optimum-intel[nncf] @ git+https://github.com/huggingface/optimum-intel.git@main
66
numpy<2.0.0; sys_platform == 'darwin'
77
einops==0.8.1 # For Qwen
88
transformers_stream_generator==0.0.5 # For Qwen
99
diffusers==0.32.2 # For image generation pipelines
1010
timm==1.0.15 # For exporting InternVL2
1111
torchvision # For visual language models
1212
transformers>=4.43 # For Whisper
13-
hf_transfer # for faster models download, should used with env var HF_HUB_ENABLE_HF_TRANSFER=1
13+
hf_transfer # for faster models download, should used with env var HF_HUB_ENABLE_HF_TRANSFER=1

tests/python_tests/samples/conftest.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,11 @@
2020
MODELS = {
2121
"TinyLlama-1.1B-Chat-v1.0": {
2222
"name": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
23-
"convert_args": []
23+
"convert_args": ['--weight-format', 'fp16']
2424
},
2525
"SmolLM-135M": {
2626
"name": "HuggingFaceTB/SmolLM-135M",
27-
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
27+
"convert_args": ['--trust-remote-code']
2828
},
2929
"SmolLM2-135M": {
3030
"name": "HuggingFaceTB/SmolLM2-135M",
@@ -36,23 +36,23 @@
3636
},
3737
"WhisperTiny": {
3838
"name": "openai/whisper-tiny",
39-
"convert_args": ['--trust-remote-code']
39+
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
4040
},
4141
"Qwen2.5-0.5B-Instruct": {
4242
"name": "Qwen/Qwen2.5-0.5B-Instruct",
43-
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
43+
"convert_args": ['--trust-remote-code']
4444
},
4545
"Qwen2-0.5B-Instruct": {
4646
"name": "Qwen/Qwen2-0.5B-Instruct",
47-
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
47+
"convert_args": ['--trust-remote-code']
4848
},
4949
"phi-1_5": {
5050
"name": "microsoft/phi-1_5",
5151
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
5252
},
5353
"TinyStories-1M": {
5454
"name": "roneneldan/TinyStories-1M",
55-
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
55+
"convert_args": ['--trust-remote-code']
5656
},
5757
"dreamlike-anime-1.0": {
5858
"name": "dreamlike-art/dreamlike-anime-1.0",
@@ -64,11 +64,11 @@
6464
},
6565
"llava-1.5-7b-hf": {
6666
"name": "llava-hf/llava-1.5-7b-hf",
67-
"convert_args": ['--trust-remote-code']
67+
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
6868
},
6969
"llava-v1.6-mistral-7b-hf": {
7070
"name": "llava-hf/llava-v1.6-mistral-7b-hf",
71-
"convert_args": ['--trust-remote-code']
71+
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
7272
},
7373
"dreamlike-anime-1.0": {
7474
"name": "dreamlike-art/dreamlike-anime-1.0",
@@ -84,11 +84,11 @@
8484
},
8585
"InternVL2-1B": {
8686
"name": "OpenGVLab/InternVL2-1B",
87-
"convert_args": ['--trust-remote-code']
87+
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
8888
},
8989
"Qwen2-VL-2B-Instruct": {
9090
"name": "Qwen/Qwen2-VL-2B-Instruct",
91-
"convert_args": ['--trust-remote-code']
91+
"convert_args": ['--trust-remote-code', '--weight-format', 'fp16']
9292
},
9393
"tiny-dummy-qwen2": {
9494
"name": "fxmarty/tiny-dummy-qwen2",

0 commit comments

Comments
 (0)