Update dependencies for image_recognition and nlp models

Signed-off-by: Sun, Xuehao <[email protected]>
onnx · Jul 1, 2024 · c05d2c6 · c05d2c6
1 parent 145472a
commit c05d2c6
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 2 deletions.
diff --git a/.azure-pipelines/scripts/models/env_setup.sh b/.azure-pipelines/scripts/models/env_setup.sh
@@ -36,3 +36,4 @@ source .azure-pipelines/scripts/change_color.sh
 $BOLD_YELLOW && echo "====== install requirements ======" && $RESET
 cd "/neural-compressor/examples/$model_src_dir"
 pip install -r requirements.txt
+pip list
diff --git a/.azure-pipelines/template/model-template.yml b/.azure-pipelines/template/model-template.yml
@@ -46,7 +46,7 @@ steps:
   - script: |
       docker exec ${{ parameters.modelContainerName }} bash -c "cd /neural-compressor/.azure-pipelines/scripts/models \
       && bash ${{ parameters.script_path }} --model=${{ parameters.modelName }} --stage='quantize'"
-    displayName: Qunatize
+    displayName: Quantize
 
   - script: |
       docker exec ${{ parameters.modelContainerName }} bash -c "cd /neural-compressor/.azure-pipelines/scripts/models \

diff --git a/examples/image_recognition/resnet50/quantization/ptq_static/requirements.txt b/examples/image_recognition/resnet50/quantization/ptq_static/requirements.txt
@@ -4,3 +4,4 @@ torch
 torchvision
 onnxruntime-extensions
 pillow>=8.2.0 # not directly required, pinned by Snyk to avoid a vulnerability
+opencv-python
diff --git a/examples/nlp/bert/quantization/ptq_dynamic/requirements.txt b/examples/nlp/bert/quantization/ptq_dynamic/requirements.txt
@@ -6,3 +6,4 @@ onnxruntime
 coloredlogs
 sympy
 onnxruntime-extensions
+scikit-learn
diff --git a/...les/nlp/huggingface_model/text_generation/llama/quantization/weight_only/requirements.txt b/...les/nlp/huggingface_model/text_generation/llama/quantization/weight_only/requirements.txt
@@ -3,7 +3,6 @@ transformers
 onnx
 onnxruntime
 onnxruntime-extensions
-opencv-python
 datasets
 optimum
 lm-eval==0.4.2