From 1d3d26a5c131ebc7cc0bf8b1994aae87ea915ed0 Mon Sep 17 00:00:00 2001 From: ayushpixis5941 <154226984+ayushpixis5941@users.noreply.github.com> Date: Mon, 8 Apr 2024 20:22:25 +0530 Subject: [PATCH 1/2] Update get_tokenlizer.py - google-bert/bert-base-uncased --- groundingdino/util/get_tokenlizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/groundingdino/util/get_tokenlizer.py b/groundingdino/util/get_tokenlizer.py index dd2d972..89f28c3 100644 --- a/groundingdino/util/get_tokenlizer.py +++ b/groundingdino/util/get_tokenlizer.py @@ -21,7 +21,7 @@ def get_tokenlizer(text_encoder_type): def get_pretrained_language_model(text_encoder_type): - if text_encoder_type == "bert-base-uncased" or (os.path.isdir(text_encoder_type) and os.path.exists(text_encoder_type)): + if text_encoder_type == "google-bert/bert-base-uncased" or (os.path.isdir(text_encoder_type) and os.path.exists(text_encoder_type)): return BertModel.from_pretrained(text_encoder_type) if text_encoder_type == "roberta-base": return RobertaModel.from_pretrained(text_encoder_type) From 3cd0d8ac2ee9ce15e931b11c63df4400e3e9d6a2 Mon Sep 17 00:00:00 2001 From: ayushpixis5941 <154226984+ayushpixis5941@users.noreply.github.com> Date: Mon, 8 Apr 2024 20:23:00 +0530 Subject: [PATCH 2/2] Update GroundingDINO_SwinT_OGC.py - google-bert/bert-base-uncased --- groundingdino/config/GroundingDINO_SwinT_OGC.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/groundingdino/config/GroundingDINO_SwinT_OGC.py b/groundingdino/config/GroundingDINO_SwinT_OGC.py index 9158d5f..7b1889f 100644 --- a/groundingdino/config/GroundingDINO_SwinT_OGC.py +++ b/groundingdino/config/GroundingDINO_SwinT_OGC.py @@ -31,7 +31,7 @@ embed_init_tgt = True dn_labelbook_size = 2000 max_text_len = 256 -text_encoder_type = "bert-base-uncased" +text_encoder_type = "google-bert/bert-base-uncased" use_text_enhancer = True use_fusion_layer = True use_checkpoint = True