From 7235e97239dc56cbd186e8793e596bf95808ae8e Mon Sep 17 00:00:00 2001 From: Emily Fertig Date: Fri, 23 Jan 2026 11:29:53 -0800 Subject: [PATCH] Fix `multiply_by_parameter_scale` type in adafactor optimizer. PiperOrigin-RevId: 860187651 --- precondition/tearfree/grafting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/precondition/tearfree/grafting.py b/precondition/tearfree/grafting.py index 72c9305..3dc70b7 100644 --- a/precondition/tearfree/grafting.py +++ b/precondition/tearfree/grafting.py @@ -81,7 +81,7 @@ class Options: skip_preconditioning_any_dim_gt: int = 4096 skip_preconditioning_rank1: bool = True min_dim_size_to_factor: int = 128 - multiply_by_parameter_scale: float = True + multiply_by_parameter_scale: bool = True clipping_threshold: float = 1.0