Junfeng5 commited on
Commit
53dc6f6
1 Parent(s): 705ddab

Update GLEE/glee/models/pixel_decoder/ops/functions/ms_deform_attn_func.py

Browse files
GLEE/glee/models/pixel_decoder/ops/functions/ms_deform_attn_func.py CHANGED
@@ -20,13 +20,14 @@ 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 maskdino/modeling/pixel_decoder/ops`\n"
27
  "\t`sh make.sh`\n"
28
  )
29
- print('use cuda deformable attention')
30
  # raise ModuleNotFoundError(info_string)
31
 
32
 
 
20
 
21
  try:
22
  import MultiScaleDeformableAttention as MSDA
23
+ print('use cuda MSDA')
24
  except ModuleNotFoundError as e:
25
  info_string = (
26
  "\n\nPlease compile MultiScaleDeformableAttention CUDA op with the following commands:\n"
27
  "\t`cd maskdino/modeling/pixel_decoder/ops`\n"
28
  "\t`sh make.sh`\n"
29
  )
30
+ print('use cpu deformable attention')
31
  # raise ModuleNotFoundError(info_string)
32
 
33