diff --git a/PixArt/models/PixArt_blocks.py b/PixArt/models/PixArt_blocks.py index bae7d87..05a9a87 100644 --- a/PixArt/models/PixArt_blocks.py +++ b/PixArt/models/PixArt_blocks.py @@ -13,7 +13,6 @@ import torch import torch.nn as nn from timm.models.vision_transformer import Mlp, Attention as Attention_ from einops import rearrange, repeat -import xformers.ops from .utils import add_decomposed_rel_pos from comfy import model_management