nikigoli commited on
Commit
e96fa12
·
verified ·
1 Parent(s): 40e3710

Added the functions import after installing msdeformable attention

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,6 @@ from __future__ import absolute_import
2
  from __future__ import print_function
3
  from __future__ import division
4
  from torch.autograd import gradcheck
5
- from functions.ms_deform_attn_func import MSDeformAttnFunction, ms_deform_attn_core_pytorch
6
  import spaces
7
  import gradio as gr
8
  import copy
@@ -47,6 +46,7 @@ subprocess.run(
47
  "pip install MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl"
48
  )
49
  )
 
50
 
51
  class AppSteps(Enum):
52
  JUST_TEXT = 1
 
2
  from __future__ import print_function
3
  from __future__ import division
4
  from torch.autograd import gradcheck
 
5
  import spaces
6
  import gradio as gr
7
  import copy
 
46
  "pip install MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl"
47
  )
48
  )
49
+ from functions.ms_deform_attn_func import MSDeformAttnFunction, ms_deform_attn_core_pytorch
50
 
51
  class AppSteps(Enum):
52
  JUST_TEXT = 1