Change last time

huggingface · Apr 8, 2024 · aec112f · aec112f
1 parent 9b2de17
commit aec112f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/state.py b/src/accelerate/state.py
@@ -183,7 +183,7 @@ def __init__(self, cpu: bool = False, **kwargs):
             self.backend = backend
             self.distributed_type = distributed_type
             use_deepspeed = False
-            if not cpu and not self.backend == "xla":
+            if not cpu and self.backend != "xla":
                 if int(os.environ.get("LOCAL_RANK", -1)) != -1:
                     # Deal with spawning deepspeed
                     if os.environ.get("ACCELERATE_USE_DEEPSPEED", "false") == "true":