Unverified Commit 345864eb authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

fix more torch.distributed imports (#12425)

* up

* unguard.
parent 35e538d4
...@@ -17,7 +17,10 @@ from dataclasses import dataclass ...@@ -17,7 +17,10 @@ from dataclasses import dataclass
from typing import Dict, List, Type, Union from typing import Dict, List, Type, Union
import torch 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 ( from ..models._modeling_parallel import (
ContextParallelConfig, ContextParallelConfig,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment