# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("callanwu/KGBART")
model = AutoModelForSeq2SeqLM.from_pretrained("callanwu/KGBART")Quick Links
We investigate two-stage graph pre-training strategies to improve BART's structure(Knowledge Graph) awareness.
- Downloads last month
- 6
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="callanwu/KGBART")