Spaces:
Runtime error
Runtime error
xichen98cn
commited on
Commit
•
1f7edfc
1
Parent(s):
87414d9
Update frozenseg/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py
Browse files
frozenseg/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py
CHANGED
@@ -18,16 +18,15 @@ import torch.nn.functional as F
|
|
18 |
from torch.autograd import Function
|
19 |
from torch.autograd.function import once_differentiable
|
20 |
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
#raise ModuleNotFoundError(info_string)
|
31 |
|
32 |
class MSDeformAttnFunction(Function):
|
33 |
@staticmethod
|
|
|
18 |
from torch.autograd import Function
|
19 |
from torch.autograd.function import once_differentiable
|
20 |
|
21 |
+
import MultiScaleDeformableAttention as MSDA
|
22 |
+
# except ModuleNotFoundError as e:
|
23 |
+
# info_string = (
|
24 |
+
# "\n\nPlease compile MultiScaleDeformableAttention CUDA op with the following commands:\n"
|
25 |
+
# "\t`cd mask2former/modeling/pixel_decoder/ops`\n"
|
26 |
+
# "\t`sh make.sh`\n"
|
27 |
+
# )
|
28 |
+
# print(info_string)
|
29 |
+
# #raise ModuleNotFoundError(info_string)
|
|
|
30 |
|
31 |
class MSDeformAttnFunction(Function):
|
32 |
@staticmethod
|