KBQA-R1

This repository hosts the released KBQA-R1 model checkpoints.

Checkpoints

• grailqa/: KBQA-R1 checkpoint for GrailQA.
• webqsp/: KBQA-R1 checkpoint for WebQSP.

Loading

from transformers import AutoModelForCausalLM, AutoTokenizer

repo_id = "unixin/kbqa-r1"
subfolder = "webqsp"  # or "grailqa"

tokenizer = AutoTokenizer.from_pretrained(repo_id, subfolder=subfolder)
model = AutoModelForCausalLM.from_pretrained(repo_id, subfolder=subfolder)

For code and training/evaluation details, see https://github.com/sunxin000/KBQA-R1.