ai4bharat
/

indictrans2-indic-en-dist-200M

text2text-generation

Model card Files Files and versions Community

pranjalchitale commited on 6 days ago

Commit

20ef3db

·

verified ·

1 Parent(s): 4000789

Update modeling_indictrans.py

Files changed (1) hide show

modeling_indictrans.py +1 -1

modeling_indictrans.py CHANGED Viewed

@@ -1656,7 +1656,7 @@ 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):

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