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
- try:
22
- import MultiScaleDeformableAttention as MSDA
23
- except ModuleNotFoundError as e:
24
- info_string = (
25
- "\n\nPlease compile MultiScaleDeformableAttention CUDA op with the following commands:\n"
26
- "\t`cd mask2former/modeling/pixel_decoder/ops`\n"
27
- "\t`sh make.sh`\n"
28
- )
29
- print(info_string)
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