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'
|
|
|
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 |
|