Fix error message in float8 FSDP utils (#1423)

fix error message in float8 fsdp utils Co-authored-by: Daniel Vega-Myhre <[email protected]>
pytorch · Dec 17, 2024 · ae28114 · ae28114
1 parent cbd7c29
commit ae28114
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()