Hamid Shojanazeri 4320528d40 clean up il y a 1 an
..
hf-text-generation-inference b3067b55dc fix typos and spelling errors il y a 1 an
README.md 3c84962685 adding checkpoint conversion il y a 1 an
chat_completion.py 557e881fcc aliginng the pad token with HF latest il y a 1 an
chat_utils.py 4767f09ecd Initial commit il y a 1 an
chats.json 4767f09ecd Initial commit il y a 1 an
checkpoint_converter_fsdp_hf.py 4320528d40 clean up il y a 1 an
inference.py ceaf6301e9 remove the unused code il y a 1 an
model_utils.py 76a187c4d2 clean up il y a 1 an
safety_utils.py 4767f09ecd Initial commit il y a 1 an
samsum_prompt.txt 4767f09ecd Initial commit il y a 1 an
vLLM_inference.py 4767f09ecd Initial commit il y a 1 an

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, PEFT and FSDP 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.