创建化身 - 无界AI

请求参数

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
回调重试规则：

1.

支持5级重试时间: 5, 10, 30, 60, 120 (单位：秒)，也就是分别会在5s、10s、30s...时进行回调重试

2.

当回调通知方返回"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"
}

示例代码

Shell

JavaScript

Java

Swift

Go

PHP

Python

HTTP

C

C#

Objective-C

Ruby

OCaml

Dart

R

curl --location --request POST 'https://gate.wujieai.net/wj-open/v1/avatar/create' \
--header 'Content-Type: application/json' \
--data-raw '{
  "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"
}'

返回响应

🟢200OK

application/json

Body

code

string <string>

响应code

可选

示例值:

200

message

string <string>

失败信息

可选

示例值:

success

data

object (AvatarCreateResponse_response)

可选

返回数据(object)

key

string <string>

化身 key

可选

示例值:

A2C90CF732D887E0E6D102C4472FF762

expected_second

integer <int32>

化身训练预估时间

可选

示例值:

12

示例

{
  "code": "200",
  "message": "success",
  "data": {
    "key": "A2C90CF732D887E0E6D102C4472FF762",
    "expected_second": 12
  }
}