This commit is contained in:
City
2023-10-28 02:58:13 +02:00
parent 7e067d2ac5
commit 5b42a090c4
-1
View File
@@ -13,7 +13,6 @@ import torch
import torch.nn as nn import torch.nn as nn
from timm.models.vision_transformer import Mlp, Attention as Attention_ from timm.models.vision_transformer import Mlp, Attention as Attention_
from einops import rearrange, repeat from einops import rearrange, repeat
import xformers.ops
from .utils import add_decomposed_rel_pos from .utils import add_decomposed_rel_pos
from comfy import model_management from comfy import model_management