fix more torch.distributed imports (#12425)

* up * unguard.
2025-10-08 10:45:39 +05:30
parent 35e538d46a
commit 345864eb85
1 changed files with 4 additions and 1 deletions
@@ -17,7 +17,10 @@ from dataclasses import dataclass
 from typing import Dict, List, Type, Union

 import torch
-import torch.distributed._functional_collectives as funcol
+
+
+if torch.distributed.is_available():
+    import torch.distributed._functional_collectives as funcol

 from ..models._modeling_parallel import (
    ContextParallelConfig,