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=20)
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=20)
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,