yucornetto commited on
Commit
6cd3df4
1 Parent(s): 63c1053

init for demo

Browse files

Change-Id: I0d93b08847007cd575b5798639f20f0565a20f8c

fcclip/modeling/pixel_decoder/ops/modules/ms_deform_attn.py CHANGED
@@ -21,7 +21,10 @@ from torch import nn
21
  import torch.nn.functional as F
22
  from torch.nn.init import xavier_uniform_, constant_
23
 
24
- from ..functions import MSDeformAttnFunction
 
 
 
25
  from ..functions.ms_deform_attn_func import ms_deform_attn_core_pytorch
26
 
27
 
 
21
  import torch.nn.functional as F
22
  from torch.nn.init import xavier_uniform_, constant_
23
 
24
+ try:
25
+ from ..functions import MSDeformAttnFunction
26
+ except:
27
+ pass
28
  from ..functions.ms_deform_attn_func import ms_deform_attn_core_pytorch
29
 
30