|
il y a 1 an | |
---|---|---|
.. | ||
hf-text-generation-inference | il y a 1 an | |
README.md | il y a 1 an | |
chat_completion.py | il y a 1 an | |
chat_utils.py | il y a 1 an | |
chats.json | il y a 1 an | |
inference.py | il y a 1 an | |
model_utils.py | il y a 1 an | |
safety_utils.py | il y a 1 an | |
samsum_prompt.txt | il y a 1 an | |
vLLM_inference.py | il y a 1 an |
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 and PEFT 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.