Spaces:
Paused
Paused
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
|
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 |
|