Skip to content

Commit 4220bf3

Browse files
committed
Fix transformers install
1 parent 2831322 commit 4220bf3

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

.github/workflows/build-test-reusable.yml

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,8 @@ jobs:
144144
run: |
145145
export DEBUG=1
146146
cd python
147-
conda run --no-capture-output -n triton pip install pybind11
148-
conda run --no-capture-output -n triton pip install --no-build-isolation -e '.[build,tests,tutorials]'
147+
pip install pybind11
148+
pip install --no-build-isolation -e '.[build,tests,tutorials]'
149149
150150
- name: Create test-triton command line
151151
run: |
@@ -195,13 +195,22 @@ jobs:
195195
echo "TRANSFORMERS_VERSION=$TRANSFORMERS_VERSION" | tee -a $GITHUB_ENV
196196
197197
- name: Install transformers
198+
if: inputs.env_manager == 'base'
198199
uses: ./.github/actions/install-dependency
199200
with:
200201
package: transformers
201202
repository: huggingface/transformers
202203
ref: ${{ env.TRANSFORMERS_VERSION }}
203204
try-tag-prefix: v
204-
env_manager: ${{ inputs.env-manager }}
205+
206+
- name: Install transformers
207+
if: inputs.env_manager != 'base'
208+
run:
209+
git clone --recursive https://github.com/huggingface/transformers
210+
cd transformers
211+
git checkout ${{ env.TRANSFORMERS_VERSION }}
212+
python setup.py bdist_wheel
213+
pip install dist/*.whl
205214

206215
- name: Run E2E test
207216
run: |

0 commit comments

Comments
 (0)