Views
No views yet
shivs28/jee_nujan_mix_v2_base1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "shivs28/jee_nujan_math_expert"
4tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
6
7# JEE problem format
8jee_prompt = '''<|problem|>
9Find the number of real solutions of the equation x³ - 3x² + 2x - 1 = 0 in the interval [0, 3].
10
11<|solution|>'''
12
13inputs = tokenizer(jee_prompt, return_tensors="pt")
14outputs = model.generate(
15 **inputs,
16 max_length=800,
17 temperature=0.1, # Low temperature for mathematical accuracy
18 do_sample=True,
19 pad_token_id=tokenizer.pad_token_id,
20 repetition_penalty=1.05
21)
22
23solution = tokenizer.decode(outputs[0], skip_special_tokens=True)
24print(solution)1complex_problem = '''<|problem|>
2In triangle ABC, if a = 7, b = 8, c = 9, find:
31. The area of triangle ABC
42. The radius of the circumscribed circle
53. The radius of the inscribed circle
6
7<|solution|>'''
8
9# Generate comprehensive solution
10inputs = tokenizer(complex_problem, return_tensors="pt")
11outputs = model.generate(
12 **inputs,
13 max_length=1200,
14 temperature=0.05, # Very low for multi-step problems
15 top_p=0.95,
16 do_sample=True,
17 pad_token_id=tokenizer.pad_token_id
18)1generation_config = {
2 "max_length": 800,
3 "temperature": 0.1,
4 "top_p": 0.95,
5 "do_sample": True,
6 "repetition_penalty": 1.05,
7 "pad_token_id": tokenizer.pad_token_id
8}1advanced_config = {
2 "max_length": 1200, # Longer for complex solutions
3 "temperature": 0.05, # Very low for accuracy
4 "top_p": 0.9,
5 "do_sample": True,
6 "repetition_penalty": 1.1,
7 "pad_token_id": tokenizer.pad_token_id
8}<|problem|> and <|solution|> tags1@model{jee_nujan_math_expert,
2 title={JEE NUJAN Math Expert: Fine-tuned Mathematics Specialist},
3 author={shivs28},
4 year={2025},
5 url={https://huggingface.co/shivs28/jee_nujan_math_expert}
6}