huggingface / huggingface/optimum-nvidia
Providing input_embeddings for generation instead of IDs
Open
- Dominant language
- Python
- Stars
- 1k
- Forks
- 103
- Avg merge
- 2m
- Merged PRs (30d)
- 1
Description
Is there a way to run Llama2 inference by providing the prompt as `inputs_embeds` (as allowed by the standard Llama2 [forward function](inputs_embeds))? Likewise, is there an easy way of accessing the model's embeddings module, such that we can manually map input id integers to embeddings?
Contributor guide
Assessment
This issue has not been assessed yet.