Spaces:
Runtime error
Runtime error
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
|