模型管理
千帆平台提供 API 接口对模型进行管理,这部分操作鉴权与 SFT 大模型调优一致,需要提供 Access Key 和 Secret Key,详见 官方文档。
目前支持的模型管理操作有:
使用前需要引用入Model类
from qianfan.resources import Model
获取模型详情
可以获得该模型的所有版本信息,需要提供模型的 id,可以从 智能云千帆控制台-模型仓库列表 获得,详细方法和返回参数字段参见 API 文档。
model_list = Model.list(model_id = 5862)
print(model_list['result']['modelVersionList'][0]['modelName'])
获取模型版本详情
可以获取某个模型版本的具体信息,需要提供模型版本 id,可以从 智能云千帆控制台-模型仓库列表 的某个模型详情中获得,详细方法和返回参数字段参见 API 文档。
model = Model.detail(model_version_id = 5659)
print(model['result']['modelName'])
训练任务发布为模型
可以将某个已完成的训练任务得到的模型发布至模型仓库中,需要提供任务 id 等信息,字段定义与返回参数字段参见 API 文档。
g = Model.publish(
is_new=True,
model_name="sdk_test_1",
version_meta={"taskId": 9220, "iterationId": 5234},
)
print(g['result']['modelId'])