Mar2Ding commited on
Commit
449d151
·
verified ·
1 Parent(s): 6959094

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +1 -0
setup.py CHANGED
@@ -6,6 +6,7 @@
6
 
7
  from setuptools import find_packages, setup
8
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
9
 
10
  # Package metadata
11
  NAME = "SAM 2"
 
6
 
7
  from setuptools import find_packages, setup
8
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension
9
+ import spaces
10
 
11
  # Package metadata
12
  NAME = "SAM 2"