Upload 7 files
Browse files- 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
|
808 |
|
809 |
def set_input_embeddings(self, value):
|
810 |
-
self.embed_tokens
|
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,
|