From 52e294acfef762edf8197cf25d0d818ba8153170 Mon Sep 17 00:00:00 2001 From: Gunasekar Date: Mon, 11 Sep 2023 21:44:15 +0000 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 0430bea..fd3f583 100644 --- a/README.md +++ b/README.md @@ -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,