diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py index 185c81958..9e2c03c65 100644 --- a/comfy/ldm/modules/attention.py +++ b/comfy/ldm/modules/attention.py @@ -5,7 +5,6 @@ from typing import Optional import torch import torch.nn.functional as F -from diffusers.models.attention_dispatch import sageattn from einops import rearrange, repeat from torch import nn, einsum