@@ -142,6 +142,7 @@ jobs:
142142 HF_TOKEN : ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
143143 # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
144144 CUBLAS_WORKSPACE_CONFIG : :16:8
145+ RUN_COMPILE : yes
145146 run : |
146147 python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
147148 -s -v -k "not Flax and not Onnx" \
@@ -180,58 +181,6 @@ jobs:
180181 pip install slack_sdk tabulate
181182 python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
182183
183- run_torch_lora_hotswap_tests :
184- name : PyTorch LoRA hotswapping with torch.compile CUDA tests
185-
186- runs-on :
187- group : aws-g4dn-2xlarge
188-
189- container :
190- image : diffusers/diffusers-pytorch-compile-cuda
191- options : --gpus 0 --shm-size "16gb" --ipc host
192-
193- steps :
194- - name : Checkout diffusers
195- uses : actions/checkout@v3
196- with :
197- fetch-depth : 2
198-
199- - name : NVIDIA-SMI
200- run : |
201- nvidia-smi
202- - name : Install dependencies
203- run : |
204- python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
205- python -m uv pip install -e [quality,test]
206- python -m uv pip install peft@git+https://github.com/huggingface/peft.git
207- pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
208- python -m uv pip install pytest-reportlog
209- - name : Environment
210- run : |
211- python utils/print_env.py
212- - name : Run torch compile tests on GPU
213- env :
214- HF_TOKEN : ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
215- RUN_COMPILE : yes
216- run : |
217- python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "hotswap" --make-reports=tests_torch_compile_hotswap tests/models/
218- - name : Failure short reports
219- if : ${{ failure() }}
220- run : cat reports/tests_torch_compile_hotswap_failures_short.txt
221-
222- - name : Test suite reports artifacts
223- if : ${{ always() }}
224- uses : actions/upload-artifact@v4
225- with :
226- name : torch_compile_hotswap_test_reports
227- path : reports
228-
229- - name : Generate Report and Notify Channel
230- if : always()
231- run : |
232- pip install slack_sdk tabulate
233- python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
234-
235184 run_big_gpu_torch_tests :
236185 name : Torch tests on big GPU
237186 strategy :
0 commit comments