ai4bharat
/

indictrans2-en-indic-1B

text2text-generation

Model card Files Files and versions Community

pranjalchitale commited on Jan 14

Commit

01f727d

·

1 Parent(s): 0468772

Fixes TieWeights

Files changed (1) hide show

modeling_indictrans.py +4 -4

modeling_indictrans.py CHANGED Viewed

@@ -1656,10 +1656,10 @@ class IndicTransForConditionalGeneration(IndicTransPreTrainedModel, GenerationMi
         self.post_init()
-    def tie_weights(self):
-        if self.config.share_decoder_input_output_embed:
-           self._tie_or_clone_weights(self.decoder.embed_tokens, self.lm_head)
     def get_encoder(self):
         return self.model.encoder

         self.post_init()
+    def tie_weights(self):
+        if self.config.share_decoder_input_output_embed:
+            self._tie_or_clone_weights(self.model.decoder.embed_tokens, self.lm_head)
     def get_encoder(self):
         return self.model.encoder