This is quantized version of
sarvamai/sarvam-1 created using llama.cpp
Sarvam-1 is a 2-billion parameter language model specifically optimized for Indian languages. It provides best in-class performance in 10 Indic languages (bn, gu, hi, kn, ml, mr, or, pa, ta, te) when compared with popular models like Gemma-2-2B and Llama-3.2-3B. It is also competitive against the much larger models like Llama-3.1-8B in these languages. More details can be found in our
release blog.
The model was trained with
NVIDIA NeMo™ Framework on the Yotta Shakti Cloud using HGX H100 systems.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained("sarvamai/sarvam-1")
5tokenizer = AutoTokenizer.from_pretrained("sarvamai/sarvam-1")
6
7# Example usage
8text = "कर्नाटक की राजधानी है:"
9inputs = tokenizer(text, return_tensors="pt")
10outputs = model.generate(**inputs, max_new_tokens=5)
11result = tokenizer.decode(outputs[0])
Sarvam non-commercial license: See the
LICENSE file