Update README.md
This commit is contained in:
parent
9efbcafbe4
commit
52e294acfe
@ -115,6 +115,9 @@ text = tokenizer.batch_decode(outputs)[0]
|
||||
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
|
||||
```bib
|
||||
@article{textbooks2,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user