unguard.

Author: sayakpaul

src/diffusers/hooks/context_parallel.py

@@ -22,12 +22,12 @@
 if torch.distributed.is_available():
     import torch.distributed._functional_collectives as funcol
 
-    from ..models._modeling_parallel import (
-        ContextParallelConfig,
-        ContextParallelInput,
-        ContextParallelModelPlan,
-        ContextParallelOutput,
-    )
+from ..models._modeling_parallel import (
+    ContextParallelConfig,
+    ContextParallelInput,
+    ContextParallelModelPlan,
+    ContextParallelOutput,
+)
 from ..utils import get_logger
 from ..utils.torch_utils import unwrap_module
 from .hooks import HookRegistry, ModelHook