Update README.md
This commit is contained in:
parent
9efbcafbe4
commit
52e294acfe
@ -115,6 +115,9 @@ text = tokenizer.batch_decode(outputs)[0]
|
|||||||
print(text)
|
print(text)
|
||||||
```
|
```
|
||||||
|
|
||||||
|
**Remark.** In the generation function, our model currently does not support beam search (`num_beams` >1) and `attention_mask' parameters.
|
||||||
|
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).
|
||||||
|
|
||||||
### Citation
|
### Citation
|
||||||
```bib
|
```bib
|
||||||
@article{textbooks2,
|
@article{textbooks2,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user