Thanmay's picture
Update README.md
19d7997 verified
metadata
language:
  - en
  - hi
  - mr
  - gu
  - ta
  - ml
license: llama2
tags:
  - multilingual
  - instruction-tuning
  - llama2

RomanSetu

This was trained as part of the paper RomanSetu: Efficiently unlocking multilingual capabilities of Large Language Models via Romanization. The codebase used to train and evaluate this model can be found at https://github.com/AI4Bharat/romansetu.

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM
model_path = "ai4bharat/romansetu-base-sft-roman"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path)