mirrors
/
GroundingDINO
mirror of https://github.com/IDEA-Research/GroundingDINO.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 3cd0d8ac2e into 856dde20ae

pull/319/merge
ayushpixis5941 2024-08-20 15:23:58 +09:00 committed by GitHub
parent 856dde20ae 3cd0d8ac2e
commit 35d28f7708
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
groundingdino/config/GroundingDINO_SwinT_OGC.py
View File

@ -31,7 +31,7 @@ dn_bbox_coef = 1.0
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

2
groundingdino/util/get_tokenlizer.py
View File

@ -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)