Views
No views yet
| Base Model | Qwen/Qwen2.5-7B |
| Training | Continual Pre-training (CPT) + Instruction Tuning (IT) |
| Domain | Nuclear Engineering |
| Languages | Korean, English |
| License | Apache 2.0 |
| Model | Multiple-Choice (EM) | Short-Answer (F1, %) | Descriptive (1–10) |
|---|---|---|---|
| Qwen2.5-7B (base) | 28 | 15.37 | 3.67 |
| AtomicGPT-Qwen2.5-7B (ours) | 37 | 18.08 | 3.94 |
| GPT-4 | 48 | 31.29 | 7.70 |
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4model_id = "KAERI-MLP/AtomicGPT-Qwen2.5-7B"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 device_map="auto",
10)
11model.eval()
12
13input_text = "Query about Nuclear (Atomic Energy)"
14input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
15outputs = model.generate(**input_ids, max_new_tokens=1024)
16print(tokenizer.decode(outputs[0]))1@article{atomicgpt2026,
2 title={AtomicGPT: A Domain-Adapted Large Language Model for Secure On-Premise Applications in Nuclear Engineering},
3 journal={Nuclear Engineering and Technology},
4 year={2026}
5}