Raghavan commited on
Commit
ba500e5
1 Parent(s): bb8ed51

Upload 7 files

Browse files
Files changed (1) hide show
  1. modeling_indictrans.py +2 -2
modeling_indictrans.py CHANGED
@@ -804,10 +804,10 @@ class IndicTransDecoder(IndicTransPreTrainedModel):
804
  self.post_init()
805
 
806
  def get_input_embeddings(self):
807
- return self.embed_tokens.word_embeddings
808
 
809
  def set_input_embeddings(self, value):
810
- self.embed_tokens.word_embeddings = value
811
 
812
  def forward(
813
  self,
 
804
  self.post_init()
805
 
806
  def get_input_embeddings(self):
807
+ return self.embed_tokens
808
 
809
  def set_input_embeddings(self, value):
810
+ self.embed_tokens = value
811
 
812
  def forward(
813
  self,