Junfeng5 commited on
Commit
730a8cb
1 Parent(s): 63412d5

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
@@ -26,6 +26,7 @@ except ModuleNotFoundError as e:
26
  "\t`cd maskdino/modeling/pixel_decoder/ops`\n"
27
  "\t`sh make.sh`\n"
28
  )
 
29
  # raise ModuleNotFoundError(info_string)
30
 
31
 
 
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