diff --git a/tools/llm_bench/requirements.txt b/tools/llm_bench/requirements.txt index f5f4a3fdeb..726df2aae4 100644 --- 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 new file mode 100644 index 0000000000..94e8944dfd --- /dev/null +++ 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 \ No newline at end of file