AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

update

Browse Source
This commit is contained in:
DN6
2025-10-08 14:35:36 +05:30
parent 5bfc7dd419
commit fb15ff526f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/models/modeling_utils.py
View File

@@ -1494,6 +1494,7 @@ class ModelMixin(torch.nn.Module, PushToHubMixin):
from ..hooks.context_parallel import apply_context_parallel
from .attention import AttentionModuleMixin
from .attention_dispatch import AttentionBackendName, _AttentionBackendRegistry
from .attention_processor import Attention, MochiAttention
if isinstance(config, ContextParallelConfig):
@@ -1509,8 +1510,6 @@ class ModelMixin(torch.nn.Module, PushToHubMixin):
# Step 1: Validate attention backend supports context parallelism if enabled
if config.context_parallel_config is not None:
from .attention_dispatch import AttentionBackendName, _AttentionBackendRegistry
for module in self.modules():
if not isinstance(module, attention_classes):
continue