Skip to content

Commit aeb4862

Browse files
committed
exclude from equality checks
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 057764d commit aeb4862

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/compressed_tensors/quantization/quant_config.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# limitations under the License.
1414

1515
from enum import Enum
16-
from typing import Any, Dict, List, Optional, Union
16+
from typing import Annotated, Any, Dict, List, Optional, Union
1717

1818
from compressed_tensors.config import CompressionFormat
1919
from compressed_tensors.quantization.quant_args import DynamicType, QuantizationArgs
@@ -142,8 +142,9 @@ class QuantizationConfig(BaseModel):
142142
quantization_status: QuantizationStatus = QuantizationStatus.INITIALIZED
143143
global_compression_ratio: Optional[float] = None
144144
ignore: Optional[List[str]] = Field(default_factory=list)
145-
run_compressed: Any = None # Dummy, unused arg for backwards compatibility
145+
# `run_compressed` is a dummy, unused arg for backwards compatibility
146146
# see: https://github.com/huggingface/transformers/pull/39324
147+
run_compressed: Annotated[Any, Field(exclude=True)] = None
147148
model_config = ConfigDict(extra="forbid")
148149

149150
def model_post_init(self, __context):

0 commit comments

Comments
 (0)