Mar2Ding commited on
Commit
ebe0ad8
·
verified ·
1 Parent(s): dc26736

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +1 -1
setup.py CHANGED
@@ -37,7 +37,7 @@ EXTRA_PACKAGES = {
37
  "dev": ["black==24.2.0", "usort==1.0.2", "ufmt==2.0.0b2"],
38
  }
39
 
40
- @spaces.GPU(duration=120)
41
  def get_extensions():
42
  srcs = ["sam2/csrc/connected_components.cu"]
43
  compile_args = {
 
37
  "dev": ["black==24.2.0", "usort==1.0.2", "ufmt==2.0.0b2"],
38
  }
39
 
40
+ # @spaces.GPU(duration=120)
41
  def get_extensions():
42
  srcs = ["sam2/csrc/connected_components.cu"]
43
  compile_args = {