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

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.