From 0dfffcd88ac46214ac8d5596c3060f87843f010f Mon Sep 17 00:00:00 2001 From: ykume Date: Sun, 11 Jun 2023 21:46:05 +0900 Subject: [PATCH] remove unnecessary import --- library/original_unet.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/library/original_unet.py b/library/original_unet.py index e22b16c0..94d11290 100644 --- a/library/original_unet.py +++ b/library/original_unet.py @@ -686,8 +686,6 @@ class CrossAttention(nn.Module): return out def forward_sdpa(self, x, context=None, mask=None): - import xformers.ops - h = self.heads q_in = self.to_q(x) context = context if context is not None else x