Hamid Shojanazeri f6152893d8 update prompts 1 tahun lalu
..
hf-text-generation-inference b3067b55dc fix typos and spelling errors 1 tahun lalu
README.md f6152893d8 update prompts 1 tahun lalu
chat_completion.py 557e881fcc aliginng the pad token with HF latest 1 tahun lalu
chat_utils.py f6152893d8 update prompts 1 tahun lalu
chats.json f6152893d8 update prompts 1 tahun lalu
checkpoint_converter_fsdp_hf.py 50e9d17045 add the default option for find the HF model_name/path from train_param.yaml 1 tahun lalu
inference.py c4e96af6ee clean up 1 tahun lalu
model_utils.py 76a187c4d2 clean up 1 tahun lalu
safety_utils.py 4767f09ecd Initial commit 1 tahun lalu
samsum_prompt.txt 4767f09ecd Initial commit 1 tahun lalu
vLLM_inference.py 4767f09ecd Initial commit 1 tahun lalu

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.

System Prompt Update

Observed Issue

We received feedback from the community on our prompt template and we are providing an update to reduce the false refusal rates seen. False refusals occur when the model incorrectly refuses to answer a question that it should, for example due to overly broad instructions to be cautious in how it provides responses.

Updated approach

Based on evaluation and analysis, we recommend the removal of the system prompt as the default setting. Pull request #104] removes the system prompt as the default option, but still provides an example to help enable experimentation for those using it.