Merge branch 'main' of github.com:salesforce/BLIP into main

mrvayd · Feb 18, 2022 · 85f8057 · 85f8057
2 parents 23af1be + 073b821
commit 85f8057
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/blip_pretrain.py b/models/blip_pretrain.py
@@ -91,7 +91,7 @@ def __init__(self,
         decoder_config.encoder_width = vision_width        
         self.text_decoder = BertLMHeadModel.from_pretrained('bert-base-uncased',config=decoder_config)    
         self.text_decoder.resize_token_embeddings(len(self.tokenizer)) 
-        tie_encoder_decoder_weights(self.text_decoder.bert,self.text_encoder,'','/attention')
+        tie_encoder_decoder_weights(self.text_encoder,self.text_decoder.bert,'','/attention')
 
 
     def forward(self, image, caption, alpha):