suriyagunasekar commited on
Commit
52e294a
1 Parent(s): 9efbcaf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -115,6 +115,9 @@ text = tokenizer.batch_decode(outputs)[0]
115
  print(text)
116
  ```
117
 
 
 
 
118
  ### Citation
119
  ```bib
120
  @article{textbooks2,
 
115
  print(text)
116
  ```
117
 
118
+ **Remark.** In the generation function, our model currently does not support beam search (`num_beams` >1) and `attention_mask' parameters.
119
+ Furthermore, in the forward pass of the model, we currently do not support outputing hidden states or attention values, using custom input embeddings (instead of the model's).
120
+
121
  ### Citation
122
  ```bib
123
  @article{textbooks2,