MoonshotAI
diff --git a/‎checkpoint_engine/distributed/base.py‎
Lines changed: 2 additions & 2 deletions b/‎checkpoint_engine/distributed/base.py‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎checkpoint_engine/distributed/hccl.py‎ ‎…eckpoint_engine/distributed/vllm_hccl.py‎checkpoint_engine/distributed/hccl.py renamed to checkpoint_engine/distributed/vllm_hccl.py b/‎checkpoint_engine/distributed/hccl.py‎ ‎…eckpoint_engine/distributed/vllm_hccl.py‎checkpoint_engine/distributed/hccl.py renamed to checkpoint_engine/distributed/vllm_hccl.py
diff --git a/‎checkpoint_engine/distributed/nccl.py‎ ‎…eckpoint_engine/distributed/vllm_nccl.py‎checkpoint_engine/distributed/nccl.py renamed to checkpoint_engine/distributed/vllm_nccl.py b/‎checkpoint_engine/distributed/nccl.py‎ ‎…eckpoint_engine/distributed/vllm_nccl.py‎checkpoint_engine/distributed/nccl.py renamed to checkpoint_engine/distributed/vllm_nccl.py
@@ -225,8 +225,8 @@ def use_backend(backend: str | None):
         return
 
     mapping = {
-        "vllm_nccl": ".nccl.DistributedNccl",
-        "vllm_hccl": ".hccl.DistributedHccl",
+        "vllm_nccl": ".vllm_nccl.DistributedNccl",
+        "vllm_hccl": ".vllm_hccl.DistributedHccl",
     }
     if backend not in mapping:
         raise ValueError(f"Unsupported custom backend: {backend}")