Update models/tag2text.py

models/tag2text.py

@@ -96,7 +96,7 @@ class Tag2Text_Caption(nn.Module):
 
         q2l_config = BertConfig.from_json_file('configs/q2l_config.json')
         q2l_config.encoder_width = vision_width
-        self.vision_multi = BertModel.from_pretrained('bert-base-uncased',config=q2l_config, add_pooling_layer=False)
+        self.vision_multi = BertModel.from_pretrained(config=q2l_config, add_pooling_layer=False)
         self.vision_multi.resize_token_embeddings(len(self.tokenizer)) 
         self.label_embed = nn.Embedding(self.num_class, q2l_config.hidden_size)
         self.fc =  GroupWiseLinear(self.num_class, num_features, bias=True)