Fix example dependency version (#1430)

VincyZhang · web-flow · commit 584ed502e586 · 2024-03-27T18:52:21.000+08:00
diff --git a/examples/huggingface/pytorch/multiple-choice/quantization/requirements.txt b/examples/huggingface/pytorch/multiple-choice/quantization/requirements.txt
@@ -5,3 +5,5 @@ torch >= 1.10.0
 transformers
 wandb
 accelerate
+schema
+
diff --git a/examples/huggingface/pytorch/question-answering/quantization/requirements.txt b/examples/huggingface/pytorch/question-answering/quantization/requirements.txt
@@ -3,3 +3,4 @@ torch >= 1.10
 transformers
 wandb
 accelerate
+schema
diff --git a/examples/huggingface/pytorch/summarization/quantization/requirements.txt b/examples/huggingface/pytorch/summarization/quantization/requirements.txt
@@ -7,3 +7,4 @@ py7zr
 torch==2.2.0
 transformers
 protobuf
+schema
diff --git a/examples/huggingface/pytorch/text-generation/quantization/run_tuning.sh b/examples/huggingface/pytorch/text-generation/quantization/run_tuning.sh
@@ -174,7 +174,7 @@ function run_tuning {
         extra_cmd=$extra_cmd" --sq --alpha ${alpha}"
         extra_cmd=$extra_cmd" --output_dir ${tuned_checkpoint}"
         extra_cmd=$extra_cmd" --trust_remote_code"
-        pip install transformers==4.33
+        pip install transformers==4.35.2
     elif [ "${topology}" = "qwen_7b" ]; then
         alpha=0.9
         model_name_or_path="Qwen/Qwen-7B"
diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -13,8 +13,8 @@ intel-tensorflow==2.14.0
 mlflow
 neural-speed
 nlpaug==1.1.9
-onnx>=1.10
-onnxruntime
+onnx==1.15.0
+onnxruntime==1.17.1
 optimum-intel
 peft==0.6.2
 py-cpuinfo
