Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("shivs28/jee_nujan_finetuned_test")
4model = AutoModelForCausalLM.from_pretrained("shivs28/jee_nujan_finetuned_test")
5
6# JEE Advanced Problem
7jee_prompt = '''<|jee_problem|>
8Problem: If f(x) = x³ - 3x² + 2x - 1, find the number of real roots of f(x) = 0 in the interval [0, 3].
9
10Solution:'''
11
12inputs = tokenizer(jee_prompt, return_tensors="pt")
13outputs = model.generate(
14 **inputs,
15 max_length=600,
16 temperature=0.1,
17 do_sample=True,
18 pad_token_id=tokenizer.pad_token_id
19)
20print(tokenizer.decode(outputs[0], skip_special_tokens=True))1# For JEE problem solving:
2generation_config = {
3 "max_length": 800,
4 "temperature": 0.1, # Very low for mathematical accuracy
5 "top_p": 0.95,
6 "do_sample": True,
7 "repetition_penalty": 1.05,
8 "pad_token_id": tokenizer.pad_token_id
9}prepare_model_for_kbit_training() setup