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