Add flax installation in daily doctest workflow (#25860)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
2
.github/workflows/doctests.yml
vendored
2
.github/workflows/doctests.yml
vendored
@@ -34,7 +34,7 @@ jobs:
|
|||||||
nvidia-smi
|
nvidia-smi
|
||||||
|
|
||||||
- name: Install transformers in edit mode
|
- name: Install transformers in edit mode
|
||||||
run: python3 -m pip install -e .
|
run: python3 -m pip install -e .[flax]
|
||||||
|
|
||||||
- name: GPU visibility
|
- name: GPU visibility
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
Reference in New Issue
Block a user