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