-
Multilingual Math Reasoning:
- Designed for solving complex math problems in English and Tulu.
-
Text Generation:
- Generates detailed and contextually accurate text responses.
-
Fine-Tuned Specializations:
- Trained on the microsoft/orca-math-word-problems-200k dataset for word problem-solving.
-
Special Token Mapping:
- Configured to use tokens for specific functions such as
<PAD> and <EOS> effectively.
-
Secure and Efficient Storage:
- Model weights are stored in the Safetensors format for secure and faster inference.
-
Large Parameter Size:
- 8.03 billion parameters enable handling complex queries and multi-turn conversations.
-
Mathematical Word Problems:
- Solve structured or unstructured math problems in natural language.
-
Conversational AI for Math:
- Engage users in interactive dialogues focused on math and logic reasoning.
-
Multilingual Support:
- Supports queries in Tulu and English, enhancing accessibility.
-
Education Tools:
- Useful in tutoring systems for math, helping students with problem-solving.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "prithivMLmods/Tulu-MathLingo-8B"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="fp16")
1query = "If a train travels 60 miles in 2 hours, what is its average speed?"
2inputs = tokenizer(query, return_tensors="pt")
3outputs = model.generate(**inputs, max_length=100)
4
5response = tokenizer.decode(outputs[0], skip_special_tokens=True)
6print("Answer:", response)