Usage

We recommend using inference engine from IndicTrans2 for inference of these models. Please execute the following block of code from the root directory:

from inference.engine import Model

model = Model(ckpt_dir, model_type="ctranslate2")

sents = [sent1, sent2,...]

# for a batch of sentences
model.batch_translate(sents, src_lang, tgt_lang)

# for a paragraph
model.translate_paragraph(text, src_lang, tgt_lang)

Citation

If you use these models, please cite the following work:

@inproceedings{gumma-etal-2025-towards,
    title = "Towards Inducing Long-Context Abilities in Multilingual Neural Machine Translation Models",
    author = "Gumma, Varun  and
      Chitale, Pranjal A  and
      Bali, Kalika",
    editor = "Chiruzzo, Luis  and
      Ritter, Alan  and
      Wang, Lu",
    booktitle = "Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)",
    month = apr,
    year = "2025",
    address = "Albuquerque, New Mexico",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2025.naacl-long.366/",
    pages = "7158--7170",
    ISBN = "979-8-89176-189-6"
}