Upload 7 files

modeling_indictrans.py

@@ -825,6 +825,12 @@ class IndicTransDecoder(IndicTransPreTrainedModel):
     def set_input_embeddings(self, value):
         self.embed_tokens = value
 
+    def get_output_embeddings(self):
+        return self.lm_head
+
+    def set_output_embeddings(self, new_embeddings):
+        self.lm_head = new_embeddings
+
     def forward(
             self,
             input_ids: Optional[torch.Tensor] = None,