Views
No views yet
transformers library.pip install transformers accelerate torch1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "krishanwalia30/Qwen3-16bit-OpenMathReasoning-Finetuned-Merged"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype="torch.float16")
7
8messages = [
9 {"role": "user", "content": "Explain the Pythagorean theorem in simple terms."},
10 {"role": "assistant", "content": "Okay, here's a simple explanation:"},
11 {"role": "user", "content": "Now, solve for the hypotenuse if a=3 and b=4."},
12]
13text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
14
15inputs = tokenizer(text, return_tensors="pt").to(model.device)
16
17outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.7, top_p=0.8, top_k=20, do_sample=True)
18print(tokenizer.decode(outputs[0], skip_special_tokens=True))per_device_train_batch_size: 2gradient_accumulation_steps: 4learning_rate: 2e-4max_steps: 30adamw_8bitlinear