From 01d7faeeb632a8da8e46016178078d267913a25d Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 2 Oct 2025 18:30:04 +0530 Subject: [PATCH] conditionally import torch distributed stuff. --- src/diffusers/models/attention_dispatch.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/diffusers/models/attention_dispatch.py b/src/diffusers/models/attention_dispatch.py index 0a2ad681237b..e1694910997a 100644 --- a/src/diffusers/models/attention_dispatch.py +++ b/src/diffusers/models/attention_dispatch.py @@ -20,7 +20,10 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Literal, Optional, Tuple, Union import torch -import torch.distributed._functional_collectives as funcol + + +if torch.distributed.is_available(): + import torch.distributed._functional_collectives as funcol from ..utils import ( get_logger,