Spaces:
Running
on
Zero
Running
on
Zero
Update setup.py
Browse files
setup.py
CHANGED
|
@@ -36,7 +36,7 @@ EXTRA_PACKAGES = {
|
|
| 36 |
"dev": ["black==24.2.0", "usort==1.0.2", "ufmt==2.0.0b2"],
|
| 37 |
}
|
| 38 |
|
| 39 |
-
|
| 40 |
def get_extensions():
|
| 41 |
srcs = ["sam2/csrc/connected_components.cu"]
|
| 42 |
compile_args = {
|
|
|
|
| 36 |
"dev": ["black==24.2.0", "usort==1.0.2", "ufmt==2.0.0b2"],
|
| 37 |
}
|
| 38 |
|
| 39 |
+
@spaces.GPU(duration=120)
|
| 40 |
def get_extensions():
|
| 41 |
srcs = ["sam2/csrc/connected_components.cu"]
|
| 42 |
compile_args = {
|