Skip to content

Commit d2cbc63

Browse files
committed
enable hotswapping tests on our nightly CI.
1 parent 05e7a85 commit d2cbc63

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

.github/workflows/nightly_tests.yml

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,24 +193,28 @@ jobs:
193193
run: |
194194
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
195195
python -m uv pip install -e [quality,test,training]
196+
python -m uv pip install peft@git+https://github.com/huggingface/peft.git
196197
- name: Environment
197198
run: |
198199
python utils/print_env.py
199-
- name: Run torch compile tests on GPU
200+
- name: Run torch hotswap + compile tests on GPU
200201
env:
201202
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
202203
RUN_COMPILE: yes
203204
run: |
204205
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "compile" --make-reports=tests_torch_compile_cuda tests/
206+
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "hotswap" --make-reports=tests_torch_hotswap_cuda tests/
205207
- name: Failure short reports
206208
if: ${{ failure() }}
207-
run: cat reports/tests_torch_compile_cuda_failures_short.txt
209+
run: |
210+
cat reports/tests_torch_compile_cuda_failures_short.txt
211+
cat reports/tests_torch_hotswap_cuda_failures_short.txt
208212
209213
- name: Test suite reports artifacts
210214
if: ${{ always() }}
211215
uses: actions/upload-artifact@v4
212216
with:
213-
name: torch_compile_test_reports
217+
name: torch_compile_hotswap_test_reports
214218
path: reports
215219

216220
run_big_gpu_torch_tests:

.github/workflows/push_tests.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ jobs:
189189
env:
190190
HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
191191
RUN_COMPILE: yes
192+
RUN_SLOW: yes
192193
run: |
193194
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "compile" --make-reports=tests_torch_compile_cuda tests/
194195
- name: Failure short reports

0 commit comments

Comments
 (0)