创建模型精调作业
更新时间:2024-04-09
功能介绍
用于创建模型精调作业。
注意事项
调用本文API,需使用安全认证AK/SK鉴权,调用流程及鉴权介绍详见SDK安装及使用流程。
调用示例
import os
import qianfan
# 使用安全认证AK/SK鉴权,通过环境变量方式初始化;替换下列示例中参数,安全认证Access Key替换your_iam_ak,Secret Key替换your_iam_sk
os.environ["QIANFAN_ACCESS_KEY"] = "your_iam_ak"
os.environ["QIANFAN_SECRET_KEY"] = "your_iam_sk"
from qianfan import resources
create_job_resp = resources.FineTune.V2.create_job(
name="job_1",
description="eb speed tasks",
model="ERNIE-Speed",
train_mode="SFT",
)
print(create_job_resp)
返回示例
QfResponse(
code=200,
headers={...},
body={
"requestId": "db9f0333-f72c-487e-b682-f1c8b6c8bdd6",
"result": {"jobId": "job-6ucxxxxx8q0"},
},
)
参数说明
请求参数
名称 | 类型 | 必填 | 描述 |
---|---|---|---|
name | string | 是 | 作业名称,取值范围:字符串长度为 [2, 20] |
description | string | 否 | 作业描述,取值范围:字符串长度为 [0, 500] |
model | string | 是 | 基础模型版本,支持对话续写类、图像生成类模型版本,说明: (1)对话续写类模型版本,该字段可选值如下: · ERNIE-Speed-8K · ERNIE-Lite-8K-0308 · ERNIE-Lite-8K-0704 · ERNIE-Lite-8K-0725 · ERNIE-Lite-8K-0922 · ERNIE-Tiny-8K · Qianfan-Chinese-Llama-2-7B-32K · Qianfan-Chinese-Llama-2-13B-v1 · Qianfan-Chinese-Llama-2-13B-v2 · Mixtral-8x7B · SQLCoder-7B · ChatGLM2-6B-32K · ChatGLM2-6B · Baichuan2-7B-Chat · Baichuan2-13B-Chat · BLOOMZ-7B · CodeLlama-7B (2)图像类模型版本,该字段可选值如下: · WENXIN-YIGE · Stable-Diffusion-XL-Base-1.0 |
train_mode | string | 是 | 调优的训练模式类型,说明: (1)当model值为ERNIE-Speed-8K,该参数可选值为以下:SFT、PostPretrain 或 DPO (2)当model值为ERNIE-Lite-8K-0308 或 ERNIE-Tiny-8K,该参数可选值为以下:SFT 或 DPO (3)ERNIE-Lite-8K-0922或Qianfan-Chinese-Llama-2-13B-v1,该参数可选值为以下:SFT 或 PostPretrain (4)其他,该参数为固定值SFT |
返回参数
名称 | 类型 | 描述 |
---|---|---|
requestId | string | 请求ID |
result | object | 请求结果 |
result说明
名称 | 类型 | 描述 |
---|---|---|
jobId | string | 作业ID |