Update modeling_rotary_indictrans.py

modeling_rotary_indictrans.py

@@ -27,6 +27,7 @@ from einops import rearrange, repeat
 from torch.amp import autocast
 from torch import einsum
 
+from transformers.generation import GenerationMixin
 from transformers.modeling_utils import PreTrainedModel
 from .configuration_rotary_indictrans import RotaryIndicTransConfig
 
@@ -1496,7 +1497,7 @@ class RotaryIndicTransModel(RotaryIndicTransPreTrainedModel):
 
 
 # Copied from transformers.models.m2m_100.modeling_m2m_100.M2M100ForConditionalGeneration->RotaryIndicTrans
-class RotaryIndicTransForConditionalGeneration(RotaryIndicTransPreTrainedModel):
+class RotaryIndicTransForConditionalGeneration(RotaryIndicTransPreTrainedModel, GenerationMixin):
     base_model_prefix = "model"
     _tied_weights_keys = None
     _label_smoothing = 0.0