Fixed make style

Author: shimizust

src/accelerate/accelerator.py

@@ -2039,11 +2039,7 @@ def _prepare_ao(self, *args):
 
         # Invariant: with FSDP2, optimizer is always passed to `prepare()` together with model
         # We only precompute scales if float8 all gather is enabled, possibly can add a flag for this later
-        if (
-            self.is_fsdp2
-            and len(optimizers) > 0
-            and self.ao_recipe_handler.config.enable_fsdp_float8_all_gather
-        ):
+        if self.is_fsdp2 and len(optimizers) > 0 and self.ao_recipe_handler.config.enable_fsdp_float8_all_gather:
             from torchao.float8 import precompute_float8_dynamic_scale_for_fsdp
 
             optimizers[0].register_step_post_hook(

src/accelerate/commands/config/cluster.py

@@ -28,6 +28,7 @@
     is_musa_available,
     is_npu_available,
     is_sdaa_available,
+    is_torchao_available,
     is_transformer_engine_available,
     is_transformers_available,
     is_xpu_available,
@@ -794,7 +795,9 @@ def get_cluster_input():
             )
             if mixed_precision == "fp8":
                 if not is_fp8_available():
-                    raise ValueError("FP8 (either torchao, Transformer Engine or MSAMP) is not installed on this machine.")
+                    raise ValueError(
+                        "FP8 (either torchao, Transformer Engine or MSAMP) is not installed on this machine."
+                    )
                 fp8_config = {}
                 fp8_config["backend"] = _ask_options(
                     "Which FP8 backend do you want to use?",
@@ -870,9 +873,9 @@ def get_cluster_input():
                         lambda x: "O1" if x == 0 else "O2",
                         default=1,
                     )
-                
+
                 elif fp8_config["backend"] == "AO":
-                    if not is_torch_ao_available():
+                    if not is_torchao_available():
                         raise ValueError("torchao was selected, but it is not installed on this machine.")
                     fp8_config["enable_fsdp_float8_all_gather"] = _ask_field(
                         "Do you want to enable FSDP2 float8 all gather? This is recommended for better performance if using FSDP2. [YES/no]: ",

src/accelerate/utils/dataclasses.py

@@ -321,7 +321,7 @@ class AORecipeKwargs(KwargsHandler):
               operations to prevent runtime errors.
             - `enable_fsdp_float8_all_gather=True`: Enables FP8 all-gather for FSDP2. This provides memory bandwidth
               savings by casting parameters before the all-gather operation, saving 50% bandwidth compared to BF16.
-            
+
             You can override these defaults by providing your own `Float8LinearConfig` instance.
         module_filter_func (`Callable`, *optional*, default to `None`):
             Optional function that must take in a module and layer name, and returns a boolean indicating whether the
@@ -338,7 +338,7 @@ def __post_init__(self):
         env_prefix = "ACCELERATE_FP8_"
         if not is_torchao_available():
             raise ImportError("TorchAO is not available. Please install it or use a different backend.")
-        
+
         if self.config is None:
             from torchao.float8 import Float8LinearConfig