Spaces:
Running
on
Zero
Running
on
Zero
derektan
commited on
Commit
·
97c5862
1
Parent(s):
65b732d
Attempt 5s GPU allocation
Browse files
Taxabind/TaxaBind/SatBind/clip_seg_tta.py
CHANGED
|
@@ -169,7 +169,7 @@ class ClipSegTTA:
|
|
| 169 |
)
|
| 170 |
|
| 171 |
# integration with ZeroGPU on hf
|
| 172 |
-
@spaces.GPU(duration=
|
| 173 |
def reset(self, sample_idx):
|
| 174 |
"""Reset the parameters & local model for the current sample."""
|
| 175 |
if self.load_model:
|
|
@@ -260,7 +260,7 @@ class ClipSegTTA:
|
|
| 260 |
# print(f"Local model reset in {(time.time()-start_time):.2f}s.")
|
| 261 |
|
| 262 |
# # integration with ZeroGPU on hf
|
| 263 |
-
@spaces.GPU(duration=
|
| 264 |
def execute_tta(
|
| 265 |
self,
|
| 266 |
patch_indices: list,
|
|
|
|
| 169 |
)
|
| 170 |
|
| 171 |
# integration with ZeroGPU on hf
|
| 172 |
+
@spaces.GPU(duration=5)
|
| 173 |
def reset(self, sample_idx):
|
| 174 |
"""Reset the parameters & local model for the current sample."""
|
| 175 |
if self.load_model:
|
|
|
|
| 260 |
# print(f"Local model reset in {(time.time()-start_time):.2f}s.")
|
| 261 |
|
| 262 |
# # integration with ZeroGPU on hf
|
| 263 |
+
@spaces.GPU(duration=5)
|
| 264 |
def execute_tta(
|
| 265 |
self,
|
| 266 |
patch_indices: list,
|