Skip to content
Merged
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
3 changes: 2 additions & 1 deletion .github/workflows/gpu_l4_golden_parity.yml
Original file line number Diff line number Diff line change
Expand Up @@ -74,8 +74,9 @@ jobs:
run: |
pip install -r requirements/ci/requirements.txt
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
pip install -e '.[testing,transformers]'
pip install -e '.[testing,transformers]' -r requirements/ci/transformers.txt
pip install --force-reinstall --no-deps -r requirements/ci/vibevoice.txt
python -c "from transformers import AutoConfig"
# PyPI's ORT 1.27+ wheel requires CUDA 13, while these runners use CUDA 12.
pip install --pre --no-deps --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ onnxruntime-gpu
pip install --no-deps onnxruntime-genai-cuda
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/gpu_l5_generation_e2e.yml
Original file line number Diff line number Diff line change
Expand Up @@ -74,8 +74,9 @@ jobs:
run: |
pip install -r requirements/ci/requirements.txt
pip install soundfile librosa ml_dtypes flatbuffers numpy packaging protobuf sympy coloredlogs
pip install -e '.[testing,transformers]'
pip install -e '.[testing,transformers]' -r requirements/ci/transformers.txt
pip install --force-reinstall --no-deps -r requirements/ci/vibevoice.txt
python -c "from transformers import AutoConfig"
# PyPI's ORT 1.27+ wheel requires CUDA 13, while these runners use CUDA 12.
pip install --pre --no-deps --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ onnxruntime-gpu
pip install --no-deps onnxruntime-genai-cuda
Expand Down
6 changes: 4 additions & 2 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -148,8 +148,9 @@ jobs:
run: |
pip install -r requirements/ci/requirements.txt
pip install onnxruntime
pip install -e '.[testing,transformers]'
pip install -e '.[testing,transformers]' -r requirements/ci/transformers.txt
pip install --force-reinstall --no-deps -r requirements/ci/vibevoice.txt
python -c "from transformers import AutoConfig"
- name: Run L3 synthetic parity tests
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
Expand Down Expand Up @@ -257,8 +258,9 @@ jobs:
run: |
pip install -r requirements/ci/requirements.txt
pip install onnxruntime
pip install -e '.[testing]'
pip install -e '.[testing]' -r requirements/ci/transformers.txt
pip install --force-reinstall --no-deps -r requirements/ci/vibevoice.txt
python -c "from transformers import AutoConfig"
- name: Run tests
run: |
pytest -n auto -m 'not integration and not arch_validation' --cov=src --cov-report=xml --cov-branch --junitxml junit.xml
Expand Down
3 changes: 2 additions & 1 deletion .github/workflows/nightly_l2.yml
Original file line number Diff line number Diff line change
Expand Up @@ -51,8 +51,9 @@ jobs:
- name: Install dependencies
run: |
pip install -r requirements/ci/requirements.txt
pip install -e '.[testing]'
pip install -e '.[testing]' -r requirements/ci/transformers.txt
pip install --force-reinstall --no-deps -r requirements/ci/vibevoice.txt
python -c "from transformers import AutoConfig"

# Run sequentially (-p no:xdist) to stay under 2GB RAM
- name: Run L2 architecture validation tests
Expand Down
3 changes: 3 additions & 0 deletions requirements/ci/transformers.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# VibeVoice support requires this source revision. Resolve its dependencies
# together with the project before installing VibeVoice with --no-deps.
transformers @ git+https://github.com/huggingface/transformers.git@640a08a597034221ca1c4fc0c129cf0118179225
6 changes: 3 additions & 3 deletions requirements/ci/vibevoice.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# VibeVoice is upstream but not yet available in a released Transformers wheel.
transformers @ git+https://github.com/huggingface/transformers.git@640a08a597034221ca1c4fc0c129cf0118179225
# Exact source used by VibeVoice Realtime L3 synthetic parity tests. CI installs
# this file with --no-deps, retaining the test environment's Transformers build.
# this file with --no-deps after resolving requirements/ci/transformers.txt.
# Upstream still declares Transformers <5; the CI reference uses the pinned
# Transformers source above rather than downgrading to that older API.
vibevoice @ git+https://github.com/microsoft/VibeVoice.git@79e516a3e20b599f137c9da03410a2a0b473b63b
Loading