@@ -234,3 +234,67 @@ jobs:
234234 with :
235235 name : pr_${{ matrix.config.report }}_test_reports
236236 path : reports
237+
238+ run_lora_tests :
239+ needs : [check_code_quality, check_repository_consistency]
240+ strategy :
241+ fail-fast : false
242+
243+ name : LoRA tests with PEFT main
244+
245+ runs-on :
246+ group : aws-general-8-plus
247+
248+ container :
249+ image : diffusers/diffusers-pytorch-cpu
250+ options : --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/
251+
252+ defaults :
253+ run :
254+ shell : bash
255+
256+ steps :
257+ - name : Checkout diffusers
258+ uses : actions/checkout@v3
259+ with :
260+ fetch-depth : 2
261+
262+ - name : Install dependencies
263+ run : |
264+ python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
265+ python -m uv pip install -e [quality,test]
266+ # TODO (sayakpaul, DN6): revisit `--no-deps`
267+ python -m pip install -U peft@git+https://github.com/huggingface/peft.git --no-deps
268+ python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
269+ pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps
270+
271+ - name : Environment
272+ run : |
273+ python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
274+ python utils/print_env.py
275+
276+ - name : Run fast PyTorch LoRA tests with PEFT
277+ run : |
278+ python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
279+ python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
280+ -s -v \
281+ --make-reports=tests_peft_main \
282+ tests/lora/
283+ python -m pytest -n 4 --max-worker-restart=0 --dist=loadfile \
284+ -s -v \
285+ --make-reports=tests_models_lora_peft_main \
286+ tests/models/ -k "lora"
287+
288+ - name : Failure short reports
289+ if : ${{ failure() }}
290+ run : |
291+ cat reports/tests_lora_failures_short.txt
292+ cat reports/tests_models_lora_failures_short.txt
293+
294+ - name : Test suite reports artifacts
295+ if : ${{ always() }}
296+ uses : actions/upload-artifact@v4
297+ with :
298+ name : pr_main_test_reports
299+ path : reports
300+
0 commit comments