minor lints

Signed-off-by: Peter St. John <[email protected]>

Author: pstjohn

src/accelerate/state.py

@@ -944,12 +944,13 @@ def __init__(
                     "Please make sure to properly initialize your accelerator via `accelerator = Accelerator()` "
                     "before using any functionality from the `accelerate` library."
                 )
-            # deepspeed handles mixed_precision using deepspeed_config
-            self._mixed_precision = (
-                "no"
-                if (self.distributed_type == DistributedType.DEEPSPEED and mixed_precision != "fp8")
-                else mixed_precision
-            )
+            # deepspeed handles mixed_precision using deepspeed_config. But we need to set it to fp8
+            # if we're using fp8.
+            if self.distributed_type == DistributedType.DEEPSPEED and mixed_precision != "fp8":
+                self._mixed_precision = "no"
+            else:
+                self._mixed_precision = mixed_precision
+
             if self.distributed_type == DistributedType.XLA and is_torch_xla_available(check_is_tpu=True):
                 if mixed_precision == "bf16":
                     if os.environ.get("ACCELERATE_DOWNCAST_BF16"):

src/accelerate/test_utils/testing.py

@@ -543,11 +543,11 @@ def require_transformer_engine(test_case):
 
 def require_transformer_engine_mxfp8(test_case):
     """
-    Decorator marking a test that requires transformers engine FP8 block scaling available. These tests are skipped
-    when transformers engine FP8 block scaling isn't available
+    Decorator marking a test that requires transformers engine MXFP8 block scaling available. These tests are skipped
+    when transformers engine MXFP8 block scaling isn't available
     """
     return unittest.skipUnless(
-        is_transformer_engine_mxfp8_available(), "test requires transformers engine FP8 block scaling"
+        is_transformer_engine_mxfp8_available(), "test requires transformers engine MXFP8 block scaling"
     )(test_case)