suriyagunasekar commited on
Commit
621f844
1 Parent(s): 7a24267

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -87,6 +87,10 @@ text = tokenizer.batch_decode(outputs)[0]
87
  print(text)
88
  ```
89
 
 
 
 
 
90
  ### Citation
91
  ```bib
92
  @article{gunasekar2023textbooks,
 
87
  print(text)
88
  ```
89
 
90
+ **Remark.** In the generation function, our model currently does not support beam search (`num_beams` >1) and `attention_mask' parameters.
91
+ Furthermore, in the forward pass of the model, we currently do not support outputing hidden states or attention values, or using custom input embeddings (instead of the model's).
92
+
93
+
94
  ### Citation
95
  ```bib
96
  @article{gunasekar2023textbooks,