inference_with_transformers_zh

使用transformers进行推理

我们提供了命令行和Web图形界面两种方式使用原生Transformers进行推理。

下面以加载Chinese-LLaMA-2-7B/Chinese-Alpaca-2-7B模型为例说明启动方式。

命令行交互形式

python scripts/inference/inference_hf.py \
    --base_model path_to_original_llama_2_hf_dir \
    --lora_model path_to_chinese_llama2_or_alpaca2_lora \
    --with_prompt \
    --interactive

如果之前已执行了merge_llama2_with_chinese_lora_low_mem.py脚本将lora权重合并，或者下载的是完整权重，那么无需再指定--lora_model，启动方式更简单：

python scripts/inference/inference_hf.py \
    --base_model path_to_merged_llama2_or_alpaca2_hf_dir \
    --with_prompt \
    --interactive

参数说明：

• --base_model {base_model} ：存放HF格式的LLaMA-2模型权重和配置文件的目录
• --lora_model {lora_model} ：中文LLaMA-2/Alpaca-2 LoRA解压后文件所在目录，也可使用🤗Model Hub模型调用名称。若不提供此参数，则只加载--base_model指定的模型
• --tokenizer_path {tokenizer_path}：存放对应tokenizer的目录。若不提供此参数，则其默认值与--lora_model相同；若也未提供--lora_model参数，则其默认值与--base_model相同
• --with_prompt：是否将输入与prompt模版进行合并。如果加载Alpaca模型，请务必启用此选项！
• --interactive：以交互方式启动，以便进行多次单轮问答（此处不是llama.cpp中的上下文对话）
• --data_file {file_name}：非交互方式启动下，按行读取file_name中的的内容进行预测
• --predictions_file {file_name}：非交互式方式下，将预测的结果以json格式写入file_name
• --use_cpu：仅使用CPU进行推理
• --gpus {gpu_ids}：指定使用的GPU设备编号，默认为0。如使用多张GPU，以逗号分隔，如0,1,2
• --alpha {alpha}：使用NTK方法拓展上下文长度的系数，可以提升可处理的输入长度。默认为1。如果不知道怎么设置，可以保持默认值，或设为"auto"
• --load_in_8bit：使用8bit方式加载模型，降低显存占用

Web图形界面交互形式

该方式将启动Web前端页面进行交互，并且支持多轮对话。除transformers之外，需要安装gradio和mdtex2html：

pip install gradio
pip install mdtex2html

启动命令如下：

python scripts/inference/gradio_demo.py \
	--base_model path_to_original_llama_2_hf_dir \
	--lora_model path_to_chinese_alpaca2_lora

如果之前已执行了merge_llama2_with_chinese_lora_low_mem.py脚本将lora权重合并，那么无需再指定--lora_model：

python scripts/inference/gradio_demo.py --base_model path_to_merged_alpaca2_hf_dir 

参数说明：

• --base_model {base_model} ：存放HF格式的LLaMA-2模型权重和配置文件的目录
• --lora_model {lora_model} ：中文Alpaca-2 LoRA解压后文件所在目录，也可使用🤗Model Hub模型调用名称。若不提供此参数，则只加载--base_model指定的模型
• --tokenizer_path {tokenizer_path}：存放对应tokenizer的目录。若不提供此参数，则其默认值与--lora_model相同；若也未提供--lora_model参数，则其默认值与--base_model相同
• --use_cpu: 仅使用CPU进行推理
• --gpus {gpu_ids}: 指定使用的GPU设备编号，默认为0。如使用多张GPU，以逗号分隔，如0,1,2
• --alpha {alpha}：使用NTK方法拓展上下文长度的系数，可以提升有效上下文长度。默认为1。如果不知道怎么设置，可以保持默认值，或设为"auto"。关于取值对性能的影响可参考#23
• --load_in_8bit：使用8bit方式加载模型
• --max_memory：多轮对话历史中存储的最大序列长度（以token计），默认1024
• --system_prompt {system_prompt}：指定system prompt，默认为alpaca-2.txt中的内容。

注意事项

• 因不同框架的解码实现细节有差异，该脚本并不能保证复现llama.cpp的解码效果
• 该脚本仅为方便快速体验用，并未对推理速度做优化
• 如在CPU上运行7B模型推理，请确保有32GB内存；如在GPU上运行7B模型推理，请确保有16GB显存