File tree Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments