"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "5f4b5794bb361ce6c6d1b31e39b97c95cff766be"
Unverified Commit cc19726f authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Tests] add requires peft decorator. (#11037)



* add requires peft decorator.

* install peft conditionally.

* conditional deps.
Co-authored-by: default avatarDN6 <dhruv.nair@gmail.com>

---------
Co-authored-by: default avatarDN6 <dhruv.nair@gmail.com>
parent be54a95b
......@@ -414,12 +414,16 @@ jobs:
config:
- backend: "bitsandbytes"
test_location: "bnb"
additional_deps: ["peft"]
- backend: "gguf"
test_location: "gguf"
additional_deps: []
- backend: "torchao"
test_location: "torchao"
additional_deps: []
- backend: "optimum_quanto"
test_location: "quanto"
additional_deps: []
runs-on:
group: aws-g6e-xlarge-plus
container:
......@@ -437,6 +441,9 @@ jobs:
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
python -m uv pip install -e [quality,test]
python -m uv pip install -U ${{ matrix.config.backend }}
if [ "${{ join(matrix.config.additional_deps, ' ') }}" != "" ]; then
python -m uv pip install ${{ join(matrix.config.additional_deps, ' ') }}
fi
python -m uv pip install pytest-reportlog
- name: Environment
run: |
......
......@@ -33,6 +33,7 @@ from diffusers.utils.testing_utils import (
numpy_cosine_similarity_distance,
require_accelerate,
require_bitsandbytes_version_greater,
require_peft_backend,
require_torch,
require_torch_gpu,
require_transformers_version_greater,
......@@ -668,6 +669,7 @@ class SlowBnb4BitFluxTests(Base4bitTests):
max_diff = numpy_cosine_similarity_distance(expected_slice, out_slice)
self.assertTrue(max_diff < 1e-3)
@require_peft_backend
def test_lora_loading(self):
self.pipeline_4bit.load_lora_weights(
hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
......
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