Skip to content

Commit d2322cd

Browse files
author
steinmetzc
committed
Fix BitNetQuantConfig to be Optional
1 parent 3db37fc commit d2322cd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/transformers/utils/quantization_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1806,8 +1806,8 @@ def __init__(
18061806
modules_to_not_convert: Optional[List] = None,
18071807
linear_class: Optional[str] = "bitlinear",
18081808
quantization_mode: Optional[str] = "offline",
1809-
use_rms_norm: bool = None,
1810-
rms_norm_eps: float = None,
1809+
use_rms_norm: Optional[bool] = False,
1810+
rms_norm_eps: Optional[float] = 1e-6,
18111811
**kwargs,
18121812
):
18131813
if linear_class not in ["bitlinear", "autobitlinear"]:

0 commit comments

Comments
 (0)