ccdv commited on
Commit
aefb85d
1 Parent(s): 248860a
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -26,7 +26,14 @@ 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=64, no_repeat_ngram_size=7)
 
 
 
 
 
 
 
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(
30
+ text,
31
+ truncation=True,
32
+ max_length=64,
33
+ no_repeat_ngram_size=7,
34
+ num_beams=2,
35
+ early_stopping=True
36
+ )
37
  ```
38
 
39
  # ccdv/lsg-bart-base-4096-pubmed