Update modeling_indictrans.py

modeling_indictrans.py

@@ -1213,7 +1213,7 @@ class IndicTransForConditionalGeneration(IndicTransPreTrainedModel):
             # move labels to the correct device to enable PP
             labels = labels.to(lm_logits.device)
             loss_fct = nn.CrossEntropyLoss()
-            masked_lm_loss = loss_fct(lm_logits.view(-1, self.config.vocab_size), labels.view(-1))
+            masked_lm_loss = loss_fct(lm_logits.view(-1, self.config.decoder_vocab_size), labels.view(-1))
 
         if not return_dict:
             output = (lm_logits,) + outputs[1:]