下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

kv_shapes

函数功能

配置每个kv的维度信息,对应底层llm.RefInputShapes配置项。

函数原型

kv_shapes(kv_shapes)

参数说明

参数名称

数据类型

取值说明

kv_shapes

List[str]

配置每个kv的维度信息。

调用示例

from llm_datadist import ModelConfig
model_config = ModelConfig()
kv_nums = 80
model_config.kv_shapes = ["1,1024" for _ in range(kv_nums)]

返回值

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词