Skip to content

GitLab

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):
@require_torch_version_greater("2.7.1")
@require_bitsandbytes_version_greater("0.45.5")
class Bnb4BitCompileTests(QuantCompileTests):
@property
def quantization_config(self):
......
tests/quantization/bnb/test_mixed_int8.py
View file @ 4c20624c
......@@ -837,6 +837,7 @@ class BaseBnb8bitSerializationTests(Base8bitTests):
@require_torch_version_greater_equal("2.6.0")
@require_bitsandbytes_version_greater("0.45.5")
class Bnb8BitCompileTests(QuantCompileTests):
@property
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