suriyagunasekar
commited on
Commit
•
621f844
1
Parent(s):
7a24267
Update README.md
Browse files
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,
|