jadechoghari commited on
Commit
efc43f7
1 Parent(s): 1c78d80

Update unet/mv_attention.py

Browse files
Files changed (1) hide show
  1. unet/mv_attention.py +1 -1
unet/mv_attention.py CHANGED
@@ -18,7 +18,7 @@ def conv_nd(dims, *args, **kwargs):
18
  raise ValueError(f"unsupported dimensions: {dims}")
19
 
20
 
21
- from attention import *
22
 
23
  try:
24
  import xformers
 
18
  raise ValueError(f"unsupported dimensions: {dims}")
19
 
20
 
21
+ from .attention import *
22
 
23
  try:
24
  import xformers