diff --git a/llm_bench/python/requirements.txt b/llm_bench/python/requirements.txt index bbeb5de89e..ab0b16db41 100644 --- a/llm_bench/python/requirements.txt +++ b/llm_bench/python/requirements.txt @@ -1,17 +1,18 @@ --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.3.0 +openvino-tokenizers~=2024.3.0 +openvino_genai~=2024.3.0 auto-gptq>=0.5.1 # for gptq pillow -torch -transformers>=4.40.0 +torch<2.5.0 +torchvision<0.20.0 +transformers>=4.40.0,<4.43.0 diffusers>=0.22.0 -#optimum is in dependency list of optimum-intel +#optimum is in dependency list of optimum-intel git+https://github.com/huggingface/optimum-intel.git@6388aeb8738b63e28fc594af84df94590e77cb9a#egg=optimum-intel -git+https://github.com/openvinotoolkit/nncf.git@develop#egg=nncf +nncf~=2.12.0 packaging psutil timm