创建化身
POST/avatar/create创建化身
请求参数
Body 参数application/json
train_image_url_list
array[string]
必需
化身训练图片url集
示例值:
[https://cdn.wujiebantu.com/ai/C837D53F842A48AEDFA7DA2976D834FE-01.png,
https://cdn.wujiebantu.com/ai/C837D53F842A48AEDFA7DA2976D834FE-01.png]
create_type_code
integer <int32>
可选
训练类型
2-个性相机基础版（化身训练）
3-Lora训练
4-个性相机高阶版,需9-15张图片
7-写真化身V3
示例值:
2
lightweight
boolean
可选
是否轻量化训练，支持使用更少的图片进行训练，仅支持化身训练 createTypeCode = 2
true：是
false：否
示例值:
false
camera_model_train_param
object (CameraModelTrainParam_request)
可选
个性相机模型训练参数(object)
resolution
integer <int32>
可选
训练图片的尺寸大小
取值范围 [128,1024]
示例值:
512
learning_rate
可选
学习率
取值范围 [0.001,0.00000001]
示例值:
0.0001
network_alpha
integer <int32>
可选
lora训练的正则化参数
取值： 16,32,64,128
示例值:
64
base_model_name
string <enum>
必需
个性相机训练基础模型
示例值:
CHILLOUTMIX_NI_PRUNED_FP32_FIX
max_train_steps
integer <int32>
可选
最大训练步数
取值范围 [600,1200]
示例值:
800
steps_per_photos
integer <int32>
可选
每张图片最大训练次数
取值范围 [100,400]
示例值:
200
gradient_accumulation_steps
integer <int32>
可选
反向传播积累步数
取值范围 [1,4]
示例值:
4
rank_lora
integer <int32>
可选
权重的特征长度
取值范围 [32,128]
示例值:
128
lora_model_train_param
object (LoraModelTrainParam_request)
可选
Lora训练相关参数(object)
resolution
integer <int32>
可选
训练图片的尺寸大小
取值范围 [128,1024]
示例值:
512
learning_rate
可选
学习率
取值范围 [0.001,0.00000001]
示例值:
0.0001
network_alpha
integer <int32>
可选
lora训练的正则化参数
取值： 16,32,64,128
示例值:
64
base_model_id
number <int64>
可选
训练基础模型id
模型信息由lora模型信息接口获取, 接口路径pro/lora/model_info
示例值:
1
epoch
integer <int32>
迭代步数
示例值:
20
dataset_repeats
integer <int32>
数据集重复次数
示例值:
5
unet_lr
可选
unetLr
取值范围[1e-8,1e-3]
示例值:
0.00005
clip_skip
integer <int32>
可选
clipSkip
取值范围[1,12]
示例值:
2
network_dim
integer <int32>
可选
networkDim
取值范围[16,32,64,128]
示例值:
64
notify_url
string <string>
可选
回调url
回调重试规则：
- 支持5级重试时间: 5, 10, 30, 60, 120 (单位：秒)，也就是分别会在5s、10s、30s...时进行回调重试
- 当回调通知方返回"success"时，我们会认为您已经成功处理回调，并终止重试
重试返回格式如下{ "key":"A2C90CF732D887E0E6D102C4472FF762", "model_fusion_name": "AVR_101_53539200", "completeTime": 1689677935, "success": true }
示例值:
www.bertha-gottlieb.io
cost_type
string <enum>
可选
超分消耗类型 INTEGRAL-积分 DURATION-时长
示例值:
INTEGRAL
示例
{
"train_image_url_list": [
"https://cdn.wujiebantu.com/ai/C837D53F842A48AEDFA7DA2976D834FE-01.png",
"https://cdn.wujiebantu.com/ai/C837D53F842A48AEDFA7DA2976D834FE-01.png"
],
"create_type_code": 2,
"lightweight": false,
"camera_model_train_param": {
"resolution": 512,
"learning_rate": 0.0001,
"network_alpha": 64,
"base_model_name": "CHILLOUTMIX_NI_PRUNED_FP32_FIX",
"max_train_steps": 800,
"steps_per_photos": 200,
"gradient_accumulation_steps": 4,
"rank_lora": 128
},
"lora_model_train_param": {
"resolution": 512,
"learning_rate": 0.0001,
"network_alpha": 64,
"base_model_id": 1,
"epoch": 20,
"dataset_repeats": 5,
"unet_lr": 0.00005,
"clip_skip": 2,
"network_dim": 64
},
"notify_url": "www.bertha-gottlieb.io",
"cost_type": "INTEGRAL"
}示例代码
返回响应
OK(200)
HTTP 状态码: 200
内容格式: JSONapplication/json
数据结构
code
string <string>
响应code
示例值:
200
message
string <string>
失败信息
示例值:
success
data
object (AvatarCreateResponse_response)
可选
返回数据(object)
key
string <string>
化身 key
示例值:
A2C90CF732D887E0E6D102C4472FF762
expected_second
integer <int32>
化身训练预估时间
示例值:
12
示例test data
{
"code": "200",
"message": "success",
"data": {
"key": "A2C90CF732D887E0E6D102C4472FF762",
"expected_second": 12
}
}最后修改时间： 6 months ago