spdraptor commited on
Commit
18cec49
·
verified ·
1 Parent(s): dfd5807

Update modules/masking_module.py

Browse files
Files changed (1) hide show
  1. modules/masking_module.py +1 -1
modules/masking_module.py CHANGED
@@ -23,7 +23,7 @@ models_dict = {
23
  SAM_CHECKPOINT = "/home/user/app/sam2_hiera_large.pt"
24
  SAM_CONFIG = "sam2_hiera_l.yaml"
25
 
26
- @spaces.GPU(duration=10)
27
  def load_sam_image_model(
28
  device: torch.device,
29
  config: str = SAM_CONFIG,
 
23
  SAM_CHECKPOINT = "/home/user/app/sam2_hiera_large.pt"
24
  SAM_CONFIG = "sam2_hiera_l.yaml"
25
 
26
+ @spaces.GPU(duration=20)
27
  def load_sam_image_model(
28
  device: torch.device,
29
  config: str = SAM_CONFIG,