Update requirements.txt and add requirements_2024.5.txt (openvinotool…

…kit#1242) Co-authored-by: Chen Peter <[email protected]> Co-authored-by: Ekaterina Aidova <[email protected]>
pavel-esir · Nov 25, 2024 · c2baad2 · c2baad2
1 parent da7a7ca
commit c2baad2
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 6 deletions.
diff --git a/tools/llm_bench/requirements.txt b/tools/llm_bench/requirements.txt
@@ -1,17 +1,17 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 numpy
 --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
-openvino
-openvino-tokenizers
-openvino_genai
+openvino~=2024.5.0
+openvino-tokenizers~=2024.5.0
+openvino_genai~=2024.5.0
 auto-gptq>=0.5.1 # for gptq
 pillow
 torch
-transformers>=4.40.0
+transformers~=4.46.0
 diffusers>=0.22.0
 #optimum is in dependency list of optimum-intel 
-git+https://github.com/huggingface/optimum-intel.git@main#egg=optimum-intel
-git+https://github.com/openvinotoolkit/nncf.git@develop#egg=nncf
+git+https://github.com/huggingface/optimum-intel.git@d35737671ec226a66c495159588b7603066e1b07##egg=optimum-intel
+nncf ~= 2.14.0
 packaging
 psutil
 timm

diff --git a/tools/llm_bench/requirements_2024.5.txt b/tools/llm_bench/requirements_2024.5.txt
@@ -0,0 +1,22 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+numpy==2.0.2
+auto_gptq==0.7.1
+pillow==11.0.0
+torch==2.5.1
+transformers==4.46.2
+diffusers==0.31.0
+openvino==2024.5.0
+openvino-tokenizers==2024.5.0
+openvino_genai==2024.5.0
+optimum~=1.23.3
+git+https://github.com/huggingface/optimum-intel.git@d35737671ec226a66c495159588b7603066e1b07#egg=optimum-intel
+nncf ~= 2.14.0
+packaging==24.1
+psutil==6.1.0
+timm==1.0.11
+tiktoken==0.8.0
+onnx==1.17.0
+einops==0.8.0
+transformers-stream-generator==0.0.5
+bitsandbytes==0.44.1
+librosa==0.10.2.post1