kv_seq_len_dim_index

产品支持情况

产品

是否支持

Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas A2 训练系列产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

x

函数功能

配置序列长度所在维度,对应底层llm.RefInputSeqLenDimIndex配置项。

函数原型

1
kv_seq_len_dim_index(kv_seq_len_dim_index)

参数说明

参数名称

数据类型

取值说明

kv_seq_len_dim_index

int

配置序列长度所在维度。

调用示例

1
2
3
from llm_datadist import ModelConfig
model_config = ModelConfig()
model_config.kv_seq_len_dim_index = 1

返回值

约束说明