Update unet/mv_attention.py

unet/mv_attention.py

@@ -18,7 +18,7 @@ def conv_nd(dims, *args, **kwargs):
     raise ValueError(f"unsupported dimensions: {dims}")
 
     
-from .attention import *
+from attention import *
 
 try:
     import xformers
@@ -364,4 +364,4 @@ if __name__ == "__main__":
         ).cuda(),
         torch.randn(n_objects, n_views, 6, 32, 32).cuda(),
     ]
-    x_post = spt_post(x, context=context)
+    x_post = spt_post(x, context=context)