add the ability to have multi-round conversation with llama (#6769)
* update llama runner to decode single token Pull Request resolved: https://github.com/pytorch/executorch/pull/6703 Right now, we don't print the generated response in the eager runner until all tokens are generated. This is not good experience as we need to wait until all tokens are generated to see the response. This PR updates it to decode each new token immediately after it is generated. ghstack-source-id: 252924039 Differential Revision: [D65578306](https://our.internmc.facebook.com/intern/diff/D65578306/) * add the ability to have multi-round conversation with llama Ad the ability to have multi-round conversations with LLM. This will be helpful for testing long context length. Differential Revision: [D65771122](https://our.internmc.facebook.com/intern/diff/D65771122/) ghstack-source-id: 252934165 Pull Request resolved: https://github.com/pytorch/executorch/pull/6758 --------- Co-authored-by: Lunwen He <lwhecser@gmail.com>
P
pytorchbot committed
623a9a61a860ed2e18364cf0715c5c898209427b
Parent: 671f9c5
Committed by GitHub <noreply@github.com>
on 11/11/2024, 10:13:34 PM