Skip to main content
Open In Colab在 GitHub 上打开

Llama.cpp

llama.cpp Python 库是一个简单的 Python 绑定,用于@ggerganov llama.cpp

此软件包提供:

  • 通过 ctypes 接口对 C API 进行低级访问。
  • 用于文本补全的高级 Python API
    • OpenAI类 API
    • LangChain兼容性
    • LlamaIndex兼容性
  • OpenAI 兼容的 Web 服务器
    • 本地 Copilot 更换
    • 函数调用支持
    • Vision API 支持
    • 多个模型
%pip install --upgrade --quiet  llama-cpp-python
from langchain_community.embeddings import LlamaCppEmbeddings
API 参考:LlamaCppEmbeddings
llama = LlamaCppEmbeddings(model_path="/path/to/model/ggml-model-q4_0.bin")
text = "This is a test document."
query_result = llama.embed_query(text)
doc_result = llama.embed_documents([text])