From ae28114b0ab1368b0465ea811e308be15dcbb9fc Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Tue, 17 Dec 2024 10:12:49 -0800 Subject: [PATCH] Fix error message in float8 FSDP utils (#1423) fix error message in float8 fsdp utils Co-authored-by: Daniel Vega-Myhre --- torchao/float8/fsdp_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchao/float8/fsdp_utils.py b/torchao/float8/fsdp_utils.py index 9fde9922af..acc4333f9c 100644 --- a/torchao/float8/fsdp_utils.py +++ b/torchao/float8/fsdp_utils.py @@ -45,7 +45,7 @@ def precompute_float8_dynamic_scale_for_fsdp(module: nn.Module) -> None: isinstance(m, Float8Linear) and m.scaling_type_weight is ScalingType.DELAYED for m in module.modules() ): - raise NotImplementedError("Only supports delayed scaling") + raise NotImplementedError("Only supports dynamic scaling") float8_linears: List[Float8Linear] = [ m for m in module.modules()