add xpu (#3163)

huggingface · Oct 31, 2024 · b0e5fd3 · b0e5fd3
1 parent 8159c98
commit b0e5fd3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/accelerate/local_sgd.py b/src/accelerate/local_sgd.py
@@ -69,6 +69,7 @@ def __init__(self, accelerator: Accelerator, model: torch.nn.Module, local_sgd_s
             DistributedType.NO,
             DistributedType.MULTI_CPU,
             DistributedType.MULTI_GPU,
+            DistributedType.MULTI_XPU,
             DistributedType.MULTI_MLU,
             DistributedType.MULTI_MUSA,
             DistributedType.MULTI_NPU,