Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "24d624a4868cdd59efe06e0d1c61f955ae238a33"
Unverified Commit 4c20624c authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[tests] annotate compilation test classes with bnb (#11715) 

annotate compilation test classes with bnb
parent 0454fbb3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/quantization/bnb/test_4bit.py
View file @ 4c20624c
...@@ -872,6 +872,7 @@ class ExtendedSerializationTest(BaseBnb4BitSerializationTests): ...@@ -872,6 +872,7 @@ class ExtendedSerializationTest(BaseBnb4BitSerializationTests):
@require_torch_version_greater("2.7.1") @require_torch_version_greater("2.7.1")
@require_bitsandbytes_version_greater("0.45.5")
class Bnb4BitCompileTests(QuantCompileTests): class Bnb4BitCompileTests(QuantCompileTests):
@property @property
def quantization_config(self): def quantization_config(self):
......
tests/quantization/bnb/test_mixed_int8.py
View file @ 4c20624c
...@@ -837,6 +837,7 @@ class BaseBnb8bitSerializationTests(Base8bitTests): ...@@ -837,6 +837,7 @@ class BaseBnb8bitSerializationTests(Base8bitTests):
@require_torch_version_greater_equal("2.6.0") @require_torch_version_greater_equal("2.6.0")
@require_bitsandbytes_version_greater("0.45.5")
class Bnb8BitCompileTests(QuantCompileTests): class Bnb8BitCompileTests(QuantCompileTests):
@property @property
def quantization_config(self): def quantization_config(self):
......
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