ccdv commited on
Commit
248860a
1 Parent(s): 14e1bcc
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("ccdv/lsg-bart-base-4096-pubmed",
26
 
27
  text = "Replace by what you want."
28
  pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer, device=0)
29
- generated_text = pipe(text, truncation=True, max_length=128)
30
  ```
31
 
32
  # ccdv/lsg-bart-base-4096-pubmed
 
26
 
27
  text = "Replace by what you want."
28
  pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer, device=0)
29
+ generated_text = pipe(text, truncation=True, max_length=64, no_repeat_ngram_size=7)
30
  ```
31
 
32
  # ccdv/lsg-bart-base-4096-pubmed