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
@@ -18,8 +18,6 @@ import torch.nn.functional as F
|
|
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')
|
|
|
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')
|