jadechoghari
commited on
Commit
•
efc43f7
1
Parent(s):
1c78d80
Update unet/mv_attention.py
Browse files- 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
|