fix

openvinotoolkit · Mar 22, 2024 · 5669d38 · 5669d38
1 parent e15b7a5
commit 5669d38
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/examples/llm_compression/openvino/tiny_llama/requirements.txt b/examples/llm_compression/openvino/tiny_llama/requirements.txt
@@ -1,5 +1,7 @@
 -c ../../../../constraints.txt
 datasets
+onnx
 openvino
 optimum-intel
+torch
 transformers
diff --git a/examples/llm_compression/openvino/tiny_llama_find_hyperparams/requirements.txt b/examples/llm_compression/openvino/tiny_llama_find_hyperparams/requirements.txt
@@ -2,6 +2,8 @@
 datasets
 numpy>=1.23.5
 openvino
-optimum-intel[openvino]>=1.13.0
+torch
+onnx
+optimum-intel>=1.13.0
 transformers
 whowhatbench @ git+https://github.com/andreyanufr/who_what_benchmark.git