|
1 year ago | |
---|---|---|
.. | ||
chat_completion | 1 year ago | |
code_llama | 1 year ago | |
hf_text_generation_inference | 1 year ago | |
vllm | 1 year ago | |
README.md | 1 year ago | |
inference.py | 1 year ago | |
samsum_prompt.txt | 1 year ago |
This folder contains inference examples for Llama 2. So far, we have provided support for three methods of inference:
inference script script provides support for Hugging Face accelerate, PEFT and FSDP fine tuned models.
vLLM_inference.py script takes advantage of vLLM's paged attention concept for low latency.
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.