代码收藏家技术教程 2024-10-16

Llama-CPP-Python 教程

项目地址:https://gitcode.com/gh_mirrors/ll/llama-cpp-python

1. 项目介绍

Llama-CPP-Python 是一个 Python 封装库，用于访问 Llama C++ 库的功能。这个封装使得在 Python 环境中可以方便地利用 Llama 的功能，如文本处理和模型交互。该项目由 Andrei Betlen 开发并维护，它允许用户通过简单的 Python 接口来使用高度优化的 C++ 实现，从而提高性能。

2. 项目快速启动

安装依赖及库

确保你的系统已经安装了 Python >= 3.10, CMake, 和相应的编译工具。然后，你可以使用 pip 来安装 Llama-CPP-Python：

pip install llama-cpp-python

如果你需要特定版本的 CUDA 支持，或者想要启用 Metal（MPS）支持，你可以使用额外的选项：

# 对于 CUDA 版本
CUDA_VERSION=cu121 pip install \
    --extra-index-url=https://abetlen.github.io/llama-cpp-python/whl/$CUDA_VERSION \
    llama-cpp-python

# 对于 Metal (MPS)
export GGML_METAL=on
pip install llama-cpp-python

运行示例

安装完成后，你可以通过下面的命令来测试 Llama-CPP-Python 是否正确安装：

import llama_cpp
print(llama_cpp.version())

运行服务器示例以实现模型服务：

python3 -m llama_cpp server --model models/7B/llama-model gguf --n_gpu_layers 35

随后可以在浏览器中访问 http://localhost:8000/docs 查看 API 文档。

3. 应用案例和最佳实践

示例：使用 Llama 进行文本处理

import llama_cpp

# 初始化 Llama 后端
llama_cpp.llama_backend_init(False)

# 加载模型参数
params = llama_cpp.llama_context_default_params()
model_path = "/path/to/your/model"
model = llama_cpp.llama_load_model_from_file(model_path.encode(), params)

# 创建上下文
ctx = llama_cpp.llama_new_context_with_model(model, params)

# 分词示例
prompt = "Q: What is the capital of France?".encode()
tokens = (llama_cpp.llama_token * int(params.n_ctx))()
n_tokens = llama_cpp.llama_tokenize(ctx, prompt, tokens, params.n_ctx)

for token in tokens[:n_tokens]:
    print(f"Token: {token}")

# 清理资源
llama_cpp.llama_free(ctx)