Update blip_pretrain.py

mrvayd · Feb 15, 2022 · 073b821 · 073b821
1 parent ad5eec3
commit 073b821
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):