Skip to content

Commit 1808558

Browse files
Update requirements.txt and add requirements_2024.4.txt (openvinotoolkit#893)
2 parents 685c20f + b15e707 commit 1808558

File tree

2 files changed

+24
-4
lines changed

2 files changed

+24
-4
lines changed

llm_bench/python/requirements.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
11
--extra-index-url https://download.pytorch.org/whl/cpu
22
numpy
33
--extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
4-
openvino
5-
openvino-tokenizers
6-
openvino_genai
4+
openvino~=2024.4.0
5+
openvino-tokenizers~=2024.4.0
6+
openvino_genai~=2024.4.0
77
auto-gptq>=0.5.1 # for gptq
88
pillow
99
torch
1010
transformers>=4.40.0
1111
diffusers>=0.22.0
1212
#optimum is in dependency list of optimum-intel
1313
git+https://github.com/huggingface/optimum-intel.git@9a8782446e394ac07283b8bd8b44916c4f297826##egg=optimum-intel
14-
git+https://github.com/openvinotoolkit/nncf.git@develop#egg=nncf
14+
nncf~=2.13.0
1515
packaging
1616
psutil
1717
timm
+20
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
--extra-index-url https://download.pytorch.org/whl/cpu
2+
numpy==1.26.4
3+
--extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
4+
auto_gptq==0.7.1
5+
pillow==10.4.0
6+
torch==2.4.1+cpu
7+
transformers==4.43.4
8+
diffusers==0.30.2
9+
openvino==2024.4.0
10+
#optimum is in dependency list of optimum-intel
11+
git+https://github.com/huggingface/optimum-intel.git@9a8782446e394ac07283b8bd8b44916c4f297826#egg=optimum-intel
12+
git+https://github.com/openvinotoolkit/nncf.git@48b356fe64124f16a67eb719d9b1653503124f8a#egg=nncf
13+
packaging==24.1
14+
psutil==6.0.0
15+
timm==1.0.9
16+
tiktoken==0.7.0
17+
onnx==1.16.1
18+
einops==0.8.0
19+
transformers-stream-generator==0.0.5
20+
bitsandbytes==0.43.3

0 commit comments

Comments
 (0)