Unverified Commit 4e54dfe9 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`Tests`/`LoRA`/`PEFT`] Test also on PEFT / transformers / accelerate latest (#5820)



* add also peft latest on peft CI

* up

* up

* up

* Update .github/workflows/pr_test_peft_backend.yml

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 6f143533
...@@ -20,20 +20,15 @@ jobs: ...@@ -20,20 +20,15 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
config: lib-versions: ["main", "latest"]
- name: LoRA
framework: lora
runner: docker-cpu
image: diffusers/diffusers-pytorch-cpu
report: torch_cpu_lora
name: ${{ matrix.config.name }} name: LoRA - ${{ matrix.lib-versions }}
runs-on: ${{ matrix.config.runner }} runs-on: docker-cpu
container: container:
image: ${{ matrix.config.image }} image: diffusers/diffusers-pytorch-cpu
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/
defaults: defaults:
...@@ -50,16 +45,19 @@ jobs: ...@@ -50,16 +45,19 @@ jobs:
run: | run: |
apt-get update && apt-get install libsndfile1-dev libgl1 -y apt-get update && apt-get install libsndfile1-dev libgl1 -y
python -m pip install -e .[quality,test] python -m pip install -e .[quality,test]
python -m pip install git+https://github.com/huggingface/accelerate.git if [ "${{ matrix.lib-version }}" == "main" ]; then
python -m pip install -U git+https://github.com/huggingface/transformers.git
python -m pip install -U git+https://github.com/huggingface/peft.git python -m pip install -U git+https://github.com/huggingface/peft.git
python -m pip install -U git+https://github.com/huggingface/transformers.git
python -m pip install -U git+https://github.com/huggingface/accelerate.git
else
python -m pip install -U peft transformers accelerate
fi
- name: Environment - name: Environment
run: | run: |
python utils/print_env.py python utils/print_env.py
- name: Run fast PyTorch LoRA CPU tests with PEFT backend - name: Run fast PyTorch LoRA CPU tests with PEFT backend
if: ${{ matrix.config.framework == 'lora' }}
run: | run: |
python -m pytest -n 2 --max-worker-restart=0 --dist=loadfile \ python -m pytest -n 2 --max-worker-restart=0 --dist=loadfile \
-s -v \ -s -v \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment