Skip to content

Commit 6776692

Browse files
committed
add requires peft decorator.
1 parent 5428046 commit 6776692

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/quantization/bnb/test_4bit.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
numpy_cosine_similarity_distance,
3434
require_accelerate,
3535
require_bitsandbytes_version_greater,
36+
require_peft_backend,
3637
require_torch,
3738
require_torch_gpu,
3839
require_transformers_version_greater,
@@ -668,6 +669,7 @@ def test_quality(self):
668669
max_diff = numpy_cosine_similarity_distance(expected_slice, out_slice)
669670
self.assertTrue(max_diff < 1e-3)
670671

672+
@require_peft_backend
671673
def test_lora_loading(self):
672674
self.pipeline_4bit.load_lora_weights(
673675
hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"

0 commit comments

Comments
 (0)