From 21185299c40a141877160dd58cf65017942f0f84 Mon Sep 17 00:00:00 2001
From: Luciano Pianese <66173155+D4idalos@users.noreply.github.com>
Date: Mon, 16 Mar 2026 23:05:44 +0100
Subject: [PATCH] Fix import for XCCl availability check

---
 src/llama_cookbook/utils/train_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/llama_cookbook/utils/train_utils.py b/src/llama_cookbook/utils/train_utils.py
index b6858422a..e1b7fe512 100644
--- a/src/llama_cookbook/utils/train_utils.py
+++ b/src/llama_cookbook/utils/train_utils.py
@@ -22,7 +22,7 @@
 from llama_cookbook.model_checkpointing import save_fsdp_model_checkpoint_full, save_model_and_optimizer_sharded, save_optimizer_checkpoint, save_peft_checkpoint, save_model_checkpoint
 from llama_cookbook.policies import fpSixteen,bfSixteen, get_llama_wrapper
 from llama_cookbook.utils.memory_utils import MemoryTrace
-from accelerate.utils import is_xpu_available, is_ccl_available
+from accelerate.utils import is_xpu_available, is_xccl_available
 from llama_cookbook.utils.flop_utils import FlopMeasure
 def set_tokenizer_params(tokenizer: LlamaTokenizer):
     tokenizer.pad_token_id = 0
@@ -428,7 +428,7 @@ def check_frozen_layers_peft_model(model):
 
 def setup():
     """Initialize the process group for distributed training"""
-    if is_ccl_available():
+    if is_xccl_available():
         # distributed training on xpus
         dist.init_process_group("ccl")
     else: