Views
No views yet


transformers library:1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "MegaScience/Qwen2.5-7B-MegaScience"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.bfloat16, # or torch.float16 if bfloat16 is not supported
10 device_map="auto"
11)
12
13messages = [
14 {"role": "user", "content": "What is the capital of France?"},
15]
16
17text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
18model_inputs = tokenizer(text, return_tensors="pt").to(model.device)
19
20generated_ids = model.generate(
21 model_inputs.input_ids,
22 max_new_tokens=256
23)
24generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
25print(generated_text)
@article{fan2025megascience,
title={MegaScience: Pushing the Frontiers of Post-Training Datasets for Science Reasoning},
author={Fan, Run-Ze and Wang, Zengzhi and Liu, Pengfei},
year={2025},
journal={arXiv preprint arXiv:2507.16812},
url={https://arxiv.org/abs/2507.16812}
}