Views
No views yet
Assamese, Bengali, Bodo, Dogri, Gujarati, English, Hindi, Kannada, Kashmiri, Konkani, Maithili, Malayalam, Manipuri, Marathi, Nepali, Odia, Punjabi, Sanskrit, Santali, Sindhi, Tamil, Telugu, Urdu1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "sarvamai/sarvam-translate"
4
5# Load tokenizer and model
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(model_name).to('cuda:0')
8
9# Translation task
10tgt_lang = "Hindi"
11input_txt = "Be the change you wish to see in the world."
12
13# Chat-style message prompt
14messages = [
15 {"role": "system", "content": f"Translate the text below to {tgt_lang}."},
16 {"role": "user", "content": input_txt}
17]
18
19# Apply chat template to structure the conversation
20text = tokenizer.apply_chat_template(
21 messages,
22 tokenize=False,
23 add_generation_prompt=True
24)
25
26# Tokenize and move input to model device
27model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
28
29# Generate the output
30generated_ids = model.generate(
31 **model_inputs,
32 max_new_tokens=1024,
33 do_sample=True,
34 temperature=0.01,
35 num_return_sequences=1
36)
37output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
38output_text = tokenizer.decode(output_ids, skip_special_tokens=True)
39
40print("Input:", input_txt)
41print("Translation:", output_text)
42vllm serve sarvamai/sarvam-translate --port 8000 --dtype bfloat16 --max-model-len 81921from openai import OpenAI
2
3# Modify OpenAI's API key and API base to use vLLM's API server.
4openai_api_key = "EMPTY"
5openai_api_base = "http://localhost:8000/v1"
6
7client = OpenAI(
8 api_key=openai_api_key,
9 base_url=openai_api_base,
10)
11
12models = client.models.list()
13model = models.data[0].id
14
15
16tgt_lang = 'Hindi'
17input_txt = 'Be the change you wish to see in the world.'
18messages = [{"role": "system", "content": f"Translate the text below to {tgt_lang}."}, {"role": "user", "content": input_txt}]
19
20
21response = client.chat.completions.create(model=model, messages=messages, temperature=0.01)
22output_text = response.choices[0].message.content
23
24print("Input:", input_txt)
25print("Translation:", output_text)1from sarvamai import SarvamAI
2client = SarvamAI()
3response = client.text.translate(
4 input="Be the change you wish to see in the world.",
5 source_language_code="en-IN",
6 target_language_code="hi-IN",
7 speaker_gender="Male",
8 model="sarvam-translate:v1",
9)