Skip to content

Commit 30cbd04

Browse files
committed
Issues adding dependencies for optimum, trying a fix
1 parent 2548cdb commit 30cbd04

File tree

2 files changed

+12
-8
lines changed

2 files changed

+12
-8
lines changed

.ci/scripts/zephyr-utils.sh

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,14 @@ setup_zephyr_et_module () {
1717
west config manifest.project-filter -- +executorch
1818
west -v update
1919
}
20+
21+
setup_optimum() {
22+
git clone https://github.com/huggingface/optimum-executorch.git
23+
cd optimum-executorch
24+
ptyhon3 -m pip install --upgrade pip
25+
python3 -m pip install '.[dev]'
26+
python3 install_dev.py
27+
python3 -m pip install torchao==0.11.0
28+
python3 -m pip install transformers==4.52.4
29+
python3 -m pip install torchcodec==0.4.0
30+
}

.github/workflows/trunk.yml

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -90,14 +90,7 @@ jobs:
9090
if [[ "${MODEL_NAME}" == "whisper_tiny" ]]; then
9191
cd $ZEPHYR_PROJ_ROOT
9292
# For whisper-tiny, we will use optimum to export the encoder and decoder ptes
93-
git clone https://github.com/huggingface/optimum-executorch.git
94-
cd optimum-executorch
95-
pip_install --upgrade pip
96-
pip_install '.[dev]'
97-
python3 install_dev.py
98-
pip_install torchao==0.11.0
99-
pip_install transformers==4.52.4
100-
pip_install torchcodec==0.4.0
93+
setup_optimum
10194
10295
optimum-cli export executorch \
10396
--model "openai/whisper-tiny" \

0 commit comments

Comments
 (0)