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

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.