remove unnecessary import

This commit is contained in:
ykume
2023-06-11 21:46:05 +09:00
parent 9e1683cf2b
commit 0dfffcd88a

View File

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