Views
No views yet
transformers library for text generation.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "sunblaze-ucb/Qwen3-14B-GRPO-MATH-1EPOCH"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11 trust_remote_code=True
12)
13model.eval()
14
15# Example using a chat-like template, typical for instruction-tuned models like Qwen.
16# Adjust prompt format as needed for your specific use case.
17messages = [
18 {"role": "user", "content": "Question: Solve the following equation: $x + 7 = 15$. Show your steps. Answer:"}
19]
20
21text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
22model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
23
24generated_ids = model.generate(
25 model_inputs.input_ids,
26 max_new_tokens=100,
27 do_sample=True,
28 temperature=0.7,
29 top_p=0.9,
30 eos_token_id=tokenizer.eos_token_id
31)
32
33generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
34print(generated_text)1@article{zhao2025learning,
2 title = {Learning to Reason without External Rewards},
3 author = {Zhao, Xuandong and Kang, Zhewei and Feng, Aosong and Levine, Sergey and Song, Dawn},
4 journal = {arXiv preprint arXiv:2505.19590},
5 year = {2025}
6}