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,
|