ge_options

函数功能

配置额外的GE配置项。

函数原型

ge_options(ge_options)

参数说明

参数名称	数据类型	取值说明
ge_options	dict[str, str]	配置GE配置项。其中ge.flowGraphMemMaxSize比较重要，表示所有KV cache占用的最大内存, 如果设置的过大，会压缩模型的可用内存，需根据实际情况指定。

参数名称

数据类型

取值说明

ge_options

dict[str, str]

配置GE配置项。

其中ge.flowGraphMemMaxSize比较重要，表示所有KV cache占用的最大内存, 如果设置的过大，会压缩模型的可用内存，需根据实际情况指定。

调用示例

from llm_datadist import LLMConfig
ge_options = {
    "ge.flowGraphMemMaxSize": "4106127360"
}
llm_config = LLMConfig()
llm_config.ge_options = ge_options

返回值

无

约束说明

无

父主题： LLMConfig