alexnasa commited on
Commit
4e9741b
·
verified ·
1 Parent(s): c8cc175

Update eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py

Browse files
eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py CHANGED
@@ -59,7 +59,8 @@ class FlorenceSAM:
59
  except:
60
  self.torch_dtype = torch.bfloat16
61
 
62
- FLORENCE2_MODEL_ID = os.getenv('FLORENCE2_MODEL_PATH', "microsoft/Florence-2-large")
 
63
  SAM2_CHECKPOINT = os.getenv('SAM2_MODEL_PATH')
64
  SAM2_CONFIG = "configs/sam2.1/sam2.1_hiera_l.yaml"
65
 
 
59
  except:
60
  self.torch_dtype = torch.bfloat16
61
 
62
+ FLORENCE2_MODEL_ID = os.getenv('FLORENCE2_MODEL_PATH')
63
+ print(f'FLORENCE2_MODEL_ID is {FLORENCE2_MODEL_ID}')
64
  SAM2_CHECKPOINT = os.getenv('SAM2_MODEL_PATH')
65
  SAM2_CONFIG = "configs/sam2.1/sam2.1_hiera_l.yaml"
66