mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-27 14:50:20 +08:00
Fix attention dispatch bug
This commit is contained in:
parent
173b1ce0ae
commit
479e20d233
@ -5,7 +5,6 @@ from typing import Optional
|
|||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
from diffusers.models.attention_dispatch import sageattn
|
|
||||||
from einops import rearrange, repeat
|
from einops import rearrange, repeat
|
||||||
from torch import nn, einsum
|
from torch import nn, einsum
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user