LocalLLM

June 26, 2026 · View on GitHub


该类在类层次结构中的位置:

[Oxy](../agent/base_oxy.md)
├── [BaseFlow](../agent/base_flow.md)
├── [BaseLLM](./base_llm.md)
│   ├── [RemoteLLM](./remote_llm.md)
│   │   ├── [HttpLLM](./http_llm.md)
│   │   └── [OpenAILLM](./openai_llm.md)
│   ├── [LocalLLM](./local_llm.md)
│   ├── [ActorLLM](./actor_llm.md)
│   ├── [LiteLLM](./lite_llm.md)
│   └── [MockLLM](./mock_llm.md)
└── [BaseTool](../tools/base_tools.md)

简介

LocalLLM 从磁盘加载 HuggingFace Transformer 模型并在本地运行推理。它需要安装 torchtransformers。模型和分词器在 init() 阶段加载,并在后续请求中复用。

参数

参数类型 / 允许值默认值描述
model_pathstr""HuggingFace 模型目录的路径。
device_mapstr"auto"模型部署的设备映射策略。
dtypestr"bfloat16"模型权重的数据类型(例如 "bfloat16""float16")。

方法

方法协程 (async)返回值用途(简述)
init()Nonemodel_path 加载模型和分词器。
_execute(oxy_request)OxyResponse构建 payload、分词、生成文本并返回结果。

继承

请参阅 BaseLLM 类了解继承的参数和方法。

用法

oxy.LocalLLM(
    name="local_llm",
    model_path="/path/to/huggingface/model",
    device_map="auto",
    dtype="bfloat16",
)