hyesunyun commited on
Commit
41dbb11
1 Parent(s): 2e58c9d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -27,6 +27,7 @@ Based on [BART Large](https://huggingface.co/transformers/model_doc/bart.html) w
27
  Format your data so that each new article or evidence to add have `<EV>` token in front with each title prefixed by `<t>` and each abstract prefixed by `<abs>`. Please have the original summary also in the same format. You can have the list of articles and original summary concatenated in any order as long as they have the correct separator tokens.
28
 
29
  ```python
 
30
  from transformers import LEDTokenizer, LEDForConditionalGeneration
31
 
32
  tokenizer = LEDTokenizer.from_pretrained("hyesunyun/update-summarization-bart-large-longformer")
 
27
  Format your data so that each new article or evidence to add have `<EV>` token in front with each title prefixed by `<t>` and each abstract prefixed by `<abs>`. Please have the original summary also in the same format. You can have the list of articles and original summary concatenated in any order as long as they have the correct separator tokens.
28
 
29
  ```python
30
+ import torch
31
  from transformers import LEDTokenizer, LEDForConditionalGeneration
32
 
33
  tokenizer = LEDTokenizer.from_pretrained("hyesunyun/update-summarization-bart-large-longformer")