try engine test

Signed-off-by: Yu Zhentao <zhentao.yu@intel.com>
intel · Mar 19, 2024 · cc35123 · cc35123
1 parent 336ef71
commit cc35123
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/.github/checkgroup.yml b/.github/checkgroup.yml
@@ -62,6 +62,7 @@ subprojects:
       - ".github/workflows/unit-test-engine.yml"
       - "requirements.txt"
       - "setup.py"
+      - intel_extension_for_transformers/transformers/**
       - "intel_extension_for_transformers/transformers/runtime/**"
       - "!intel_extension_for_transformers/transformers/runtime/kernels/**"
       - "!intel_extension_for_transformers/transformers/runtime/third_party/**"

diff --git a/.github/workflows/llm-test.yml b/.github/workflows/llm-test.yml
@@ -53,6 +53,7 @@ jobs:
           conda activate llm-test || source activate llm-test
           pip install build --upgrade
           pip install -r requirements.txt
+          pip install torch==2.0 --extra-index-url https://download.pytorch.org/whl/cpu
           python setup.py sdist bdist_wheel
           pip install dist/intel_extension_for_transformers*.whl
           pip list

diff --git a/intel_extension_for_transformers/transformers/runtime/test/pytest/requirements.txt b/intel_extension_for_transformers/transformers/runtime/test/pytest/requirements.txt
@@ -1,4 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
+accelerate
 datasets
 intel-extension-for-pytorch==2.0.0
 intel-tensorflow==2.12.0