pruhtopia/falcon-toc-generation
Viewer • Updated • 599 • 5
How to use pruhtopia/bart-toc-generation with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("pruhtopia/bart-toc-generation")
model = AutoModelForSeq2SeqLM.from_pretrained("pruhtopia/bart-toc-generation")loss: 0.20580817759037018
rouge1: 54.3657
rouge2: 43.8004
rougeL: 50.056
rougeLsum: 53.9699
gen_len: 116.6083
runtime: 450.7126
samples_per_second: 0.266
steps_per_second: 0.033
: 3.0