Fix attention dispatch bug

This commit is contained in:
doctorpangloss 2025-08-26 16:40:45 -07:00
parent 173b1ce0ae
commit 479e20d233

View File

@ -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