Skip to content

Commit 0f96284

Browse files
authored
suppress pydantic warning in legacy quantization modifier (#2340) (#2341)
1 parent ffa3852 commit 0f96284

File tree

1 file changed

+4
-0
lines changed
  • src/sparseml/modifiers/quantization_legacy

1 file changed

+4
-0
lines changed

src/sparseml/modifiers/quantization_legacy/base.py

+4
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414

1515
from typing import Any, Dict, List, Optional
1616

17+
from pydantic import ConfigDict
18+
1719
from sparseml.core import Event, Modifier
1820

1921

@@ -81,6 +83,8 @@ class LegacyQuantizationModifier(Modifier):
8183
post_oneshot_calibration: Optional[bool] = False
8284
strict: bool = True
8385

86+
model_config = ConfigDict(protected_namespaces=())
87+
8488
def __init__(self, **kwargs):
8589
super().__init__(**kwargs)
8690
if self.model_fuse_fn_kwargs is None:

0 commit comments

Comments
 (0)