How to use ccdv/lsg-bart-base-4096 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="ccdv/lsg-bart-base-4096", trust_remote_code=True)
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("ccdv/lsg-bart-base-4096", trust_remote_code=True) model = AutoModelForSeq2SeqLM.from_pretrained("ccdv/lsg-bart-base-4096", trust_remote_code=True)
This leads to an error during training.
Hi @Agniva Made the script compatible with a recent transformers version
Thank you
· Sign up or log in to comment
