Update requirements.txt and add requirements_2024.4.txt (#893)

openvinotoolkit · Sep 25, 2024 · 1808558 · 1808558
2 parents 685c20f + b15e707
commit 1808558
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/llm_bench/python/requirements.txt b/llm_bench/python/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.4.0
+openvino-tokenizers~=2024.4.0
+openvino_genai~=2024.4.0
 auto-gptq>=0.5.1 # for gptq
 pillow
 torch
 transformers>=4.40.0
 diffusers>=0.22.0
 #optimum is in dependency list of optimum-intel 
 git+https://github.com/huggingface/optimum-intel.git@9a8782446e394ac07283b8bd8b44916c4f297826##egg=optimum-intel
-git+https://github.com/openvinotoolkit/nncf.git@develop#egg=nncf
+nncf~=2.13.0
 packaging
 psutil
 timm

diff --git a/llm_bench/python/requirements_2024.4.txt b/llm_bench/python/requirements_2024.4.txt
@@ -0,0 +1,20 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+numpy==1.26.4
+--extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
+auto_gptq==0.7.1
+pillow==10.4.0
+torch==2.4.1+cpu
+transformers==4.43.4
+diffusers==0.30.2
+openvino==2024.4.0
+#optimum is in dependency list of optimum-intel
+git+https://github.com/huggingface/optimum-intel.git@9a8782446e394ac07283b8bd8b44916c4f297826#egg=optimum-intel
+git+https://github.com/openvinotoolkit/nncf.git@48b356fe64124f16a67eb719d9b1653503124f8a#egg=nncf
+packaging==24.1
+psutil==6.0.0
+timm==1.0.9
+tiktoken==0.7.0
+onnx==1.16.1
+einops==0.8.0
+transformers-stream-generator==0.0.5
+bitsandbytes==0.43.3