huggingface / huggingface/optimum-nvidia

Providing input_embeddings for generation instead of IDs

Open
#129 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.