百度智能云文心大模型API ERNIE-ViLG模型精调快速开始
文档简介：
邀测申请：
目前ERNIE-ViLG的模型精调处于邀测阶段，您可以通过填写申请表单获取邀测资格：
申请表单：https://wenxin.baidu.com/wenxin/apply3
模型精调基本步骤：
获取API Key。
【百度智能云】文心一言
咨询热线：400-826-7010，为您提供专业的售前咨询，让您快速了解云产品，助您轻松上云！
微信咨询
免费试用、价格特惠
邀测申请
目前ERNIE-ViLG的模型精调处于邀测阶段，您可以通过填写申请表单获取邀测资格：
申请表单：https://wenxin.baidu.com/wenxin/apply3
模型精调基本步骤
获取API Key
您可以登录文心大模型API服务平台个人中心获取您的专属API Key（AK）和Secret Key（SK），查看您的AK/SK。请注意保护您的密钥信息，避免泄露。您可以通过删除已泄露的密钥来保护您的账户安全。
安装客户端
安装文心大模型API客户端wenxin-api
pip install --upgrade wenxin-api
接口调用
您可以通过以下代码使用您的数据集进行模型精调（将上一步骤申请的AK、SK填入下方代码的your ak和your sk中）。
注意为了区分ERNIE 3.0模型精调，所有类方法需要传api_type字段，实例方法不需要
- 模型精调基本步骤
# -*- coding: utf-8 -* import time import wenxin_api from wenxin_api import Dataset,
Model, Train from wenxin_api.tasks.text_to_image_training import TextToImageTuned from
wenxin_api.const import TASK_STATE_TRAIN_SUCCESS, TYPE_TEXT_TO_IMAGE wenxin_api.ak = "your ak" wenxin_api.sk = "your sk" # 上传用户自定义数据集
local_file_path = "local_file_path" dataset = Dataset.create(local_file_path, need_check=False,
api_type=TYPE_TEXT_TO_IMAGE) # 创建训练任务 # 注意请求频率，过于频繁的请求会被限制qps sleep_time
= 20 train_task = Train.create(train_datasets=[dataset], step=200, api_type=TYPE_TEXT_TO_IMAGE)
while train_task.status != TASK_STATE_TRAIN_SUCCESS: train_task.update() print("task:", train_task)
time.sleep(sleep_time) model = Model.retrieve(task_id=train_task.id, api_type=TYPE_TEXT_TO_IMAGE)
input_text = "戴墨镜的猫[c]" result = TextToImageTuned.create(text=input_text, style="探索无限",
resolution="1024*1024", model=model) print(result)
其中常见的状态码可以参考：常见状态码
