sekyonda 822b37bbc0 Update README.md 1 anno fa
..
hf-text-generation-inference 4767f09ecd Initial commit 1 anno fa
README.md 822b37bbc0 Update README.md 1 anno fa
chat_completion.py 557e881fcc aliginng the pad token with HF latest 1 anno fa
chat_utils.py 4767f09ecd Initial commit 1 anno fa
chats.json 4767f09ecd Initial commit 1 anno fa
inference.py 7ec390bfc8 aliging special tokens in toeknizer with HF latest 1 anno fa
model_utils.py 4767f09ecd Initial commit 1 anno fa
safety_utils.py 4767f09ecd Initial commit 1 anno fa
samsum_prompt.txt 4767f09ecd Initial commit 1 anno fa
vLLM_inference.py 4767f09ecd Initial commit 1 anno fa

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.