Elias James Howell b3067b55dc fix typos and spelling errors 1 ano atrás
..
hf-text-generation-inference b3067b55dc fix typos and spelling errors 1 ano atrás
README.md 822b37bbc0 Update README.md 1 ano atrás
chat_completion.py 557e881fcc aliginng the pad token with HF latest 1 ano atrás
chat_utils.py 4767f09ecd Initial commit 1 ano atrás
chats.json 4767f09ecd Initial commit 1 ano atrás
inference.py 7ec390bfc8 aliging special tokens in toeknizer with HF latest 1 ano atrás
model_utils.py 4767f09ecd Initial commit 1 ano atrás
safety_utils.py 4767f09ecd Initial commit 1 ano atrás
samsum_prompt.txt 4767f09ecd Initial commit 1 ano atrás
vLLM_inference.py 4767f09ecd Initial commit 1 ano atrás

README.md

Inference

This folder contains inference examples for Llama 2. So far, we have provided support for three methods of inference:

  1. inference script script provides support for Hugging Face accelerate and PEFT fine tuned models.

  2. vLLM_inference.py script takes advantage of vLLM's paged attention concept for low latency.

  3. The hf-text-generation-inference folder contains information on Hugging Face Text Generation Inference (TGI).

For more in depth information on inference including inference safety checks and examples, see the inference documentation here.