Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,10 @@ classifiers = [
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Operating System :: Linux",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.12",
"Topic :: Scientific/Engineering :: Artificial Intelligence for Inference Accelerator",
]
requires-python = ">=3.8,<3.11"
requires-python = ">=3.8,<3.13"
dependencies = [
"transformers==4.55.0",
"huggingface-hub==0.34.0",
Expand All @@ -44,6 +44,7 @@ dependencies = [
"torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp38-cp38-linux_x86_64.whl ; python_version=='3.8' and platform_machine=='x86_64'",
"torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'",
"torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'",
"torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl ; python_version=='3.12' and platform_machine=='x86_64'",
]

[project.optional-dependencies]
Expand Down
42 changes: 21 additions & 21 deletions scripts/Jenkinsfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,15 +17,15 @@ pipeline {
sudo docker exec ${BUILD_TAG} bash -c "
cd /efficient-transformers &&
apt update &&
apt install -y python3.10-venv &&
python3.10 -m venv preflight_qeff &&
DEBIAN_FRONTEND=noninteractive apt install -y tzdata python3.12-venv python3.12-dev build-essential &&
python3.12 -m venv preflight_qeff &&
. preflight_qeff/bin/activate &&
pip install --upgrade pip setuptools &&
pip install .[test] &&
pip install junitparser pytest-xdist &&
pip install librosa==0.10.2 soundfile==0.13.1 && #packages needed to load example for whisper testing
pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.22.0+cpu einops==0.8.1 && #packages to load VLMs
pip install /opt/qti-aic/integrations/torch_qaic/py310/torch_qaic-0.1.0-cp310-cp310-linux_x86_64.whl && # For finetuning tests
#pip install /opt/qti-aic/integrations/torch_qaic/py310/torch_qaic-0.1.0-cp310-cp310-linux_x86_64.whl && # For finetuning tests
rm -rf QEfficient"
'''
}
Expand Down Expand Up @@ -161,23 +161,23 @@ pipeline {
// }
// }
// }
stage('Finetune CLI Tests') {
steps {
timeout(time: 5, unit: 'MINUTES') {
sh '''
sudo docker exec ${BUILD_TAG} bash -c "
cd /efficient-transformers &&
. preflight_qeff/bin/activate &&
mkdir -p $PWD/cli_qaic_finetuning &&
export TOKENIZERS_PARALLELISM=false &&
export QEFF_HOME=$PWD/cli_qaic_finetuning &&
pytest tests -m '(cli) and (on_qaic) and (not qnn) and (not multimodal) and (finetune)' --ignore tests/vllm --junitxml=tests/tests_log_finetune.xml &&
junitparser merge tests/tests_log_finetune.xml tests/tests_log.xml &&
deactivate"
'''
}
}
}
// stage('Finetune CLI Tests') {
// steps {
// timeout(time: 5, unit: 'MINUTES') {
// sh '''
// sudo docker exec ${BUILD_TAG} bash -c "
// cd /efficient-transformers &&
// . preflight_qeff/bin/activate &&
// mkdir -p $PWD/cli_qaic_finetuning &&
// export TOKENIZERS_PARALLELISM=false &&
// export QEFF_HOME=$PWD/cli_qaic_finetuning &&
// pytest tests -m '(cli) and (on_qaic) and (not qnn) and (not multimodal) and (finetune)' --ignore tests/vllm --junitxml=tests/tests_log_finetune.xml &&
// junitparser merge tests/tests_log_finetune.xml tests/tests_log.xml &&
// deactivate"
// '''
// }
// }
// }
}

post {
Expand Down Expand Up @@ -216,4 +216,4 @@ pipeline {
deleteDir()
}
}
}
}