Update tests/quantization/bnb/test_mixed_int8.py

sayakpaul · DN6 · web-flow · commit 6d2a38ffa791 · 2025-01-15T08:54:40.000+05:30
Co-authored-by: Dhruv Nair &lt;dhruv.nair@gmail.com&gt;
diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py
@@ -512,7 +512,7 @@ def test_quality(self):
         self.assertTrue(max_diff < 1e-3)
 
     @require_peft_version_greater("0.14.0")
-    def test_lora_loading_works(self):
+    def test_lora_loading(self):
         self.pipeline_8bit.load_lora_weights(
             hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
         )

Original file line number	Diff line number	Diff line change
`@@ -512,7 +512,7 @@ def test_quality(self):`
`512`	`512`	`self.assertTrue(max_diff < 1e-3)`
`513`	`513`
`514`	`514`	`@require_peft_version_greater("0.14.0")`
`515`		`- def test_lora_loading_works(self):`
	`515`	`+ def test_lora_loading(self):`
`516`	`516`	`self.pipeline_8bit.load_lora_weights(`
`517`	`517`	`hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"`
`518`	`518`	`)`