bart_scitldr

作者 vadis · 下载 1 · 点赞 0 · 更新 2023-10-31

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large")
model = AutoModelForSeq2SeqLM.from_pretrained("vadis/bart_scitldr", use_auth_token=True)

text = "Abstract of a paper."
batch = tok(text, return_tensors="pt")
generated_ids = model.generate(batch["input_ids"])
print(tok.batch_decode(generated_ids, skip_special_tokens=True))

模型信息

该模型卡暂无预览图和文字说明。以下信息来自 Hugging Face。

作者 vadis
类型—
基础模型—
标签—
下载1
点赞0
更新时间2023-10-31
Hugging Face https://huggingface.co/vadis/bart_scitldr