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
|
@@ -18,6 +18,8 @@ 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 |
print('use cuda MSDA')
|
|
|
|
| 18 |
from torch.autograd import Function
|
| 19 |
from torch.autograd.function import once_differentiable
|
| 20 |
|
| 21 |
+
import MultiScaleDeformableAttention as MSDA
|
| 22 |
+
|
| 23 |
try:
|
| 24 |
import MultiScaleDeformableAttention as MSDA
|
| 25 |
print('use cuda MSDA')
|