File tree 2 files changed +9
-0
lines changed
2 files changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -414,12 +414,16 @@ jobs:
414
414
config :
415
415
- backend : " bitsandbytes"
416
416
test_location : " bnb"
417
+ additional_deps : ["peft"]
417
418
- backend : " gguf"
418
419
test_location : " gguf"
420
+ additional_deps : []
419
421
- backend : " torchao"
420
422
test_location : " torchao"
423
+ additional_deps : []
421
424
- backend : " optimum_quanto"
422
425
test_location : " quanto"
426
+ additional_deps : []
423
427
runs-on :
424
428
group : aws-g6e-xlarge-plus
425
429
container :
@@ -437,6 +441,9 @@ jobs:
437
441
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
438
442
python -m uv pip install -e [quality,test]
439
443
python -m uv pip install -U ${{ matrix.config.backend }}
444
+ if [ "${{ join(matrix.config.additional_deps, ' ') }}" != "" ]; then
445
+ python -m uv pip install ${{ join(matrix.config.additional_deps, ' ') }}
446
+ fi
440
447
python -m uv pip install pytest-reportlog
441
448
- name : Environment
442
449
run : |
Original file line number Diff line number Diff line change 33
33
numpy_cosine_similarity_distance ,
34
34
require_accelerate ,
35
35
require_bitsandbytes_version_greater ,
36
+ require_peft_backend ,
36
37
require_torch ,
37
38
require_torch_gpu ,
38
39
require_transformers_version_greater ,
@@ -668,6 +669,7 @@ def test_quality(self):
668
669
max_diff = numpy_cosine_similarity_distance (expected_slice , out_slice )
669
670
self .assertTrue (max_diff < 1e-3 )
670
671
672
+ @require_peft_backend
671
673
def test_lora_loading (self ):
672
674
self .pipeline_4bit .load_lora_weights (
673
675
hf_hub_download ("ByteDance/Hyper-SD" , "Hyper-FLUX.1-dev-8steps-lora.safetensors" ), adapter_name = "hyper-sd"
You can’t perform that action at this time.
0 commit comments