88 lines
1.8 KiB
JSON
88 lines
1.8 KiB
JSON
{
|
|
"model": "dev",
|
|
"seed": 42,
|
|
"steps": 2,
|
|
"guidance": 3.0,
|
|
"quantize": 4,
|
|
"width": 128,
|
|
"height": 128,
|
|
"training_loop": {
|
|
"num_epochs": 1,
|
|
"batch_size": 1
|
|
},
|
|
"optimizer": {
|
|
"name": "AdamW",
|
|
"learning_rate": 1e-4
|
|
},
|
|
"save": {
|
|
"output_path": "tests/dreambooth/tmp",
|
|
"checkpoint_frequency": 1
|
|
},
|
|
"instrumentation": {
|
|
"plot_frequency": 1,
|
|
"generate_image_frequency": 50000,
|
|
"validation_prompt": "photo of sks dog"
|
|
},
|
|
"lora_layers": {
|
|
"transformer_blocks": {
|
|
"block_range": {
|
|
"start": 12,
|
|
"end": 13
|
|
},
|
|
"layer_types": [
|
|
"attn.to_q",
|
|
"attn.to_k",
|
|
"attn.to_v",
|
|
"attn.add_q_proj",
|
|
"attn.add_k_proj",
|
|
"attn.add_v_proj",
|
|
"attn.to_out",
|
|
"attn.to_add_out",
|
|
"ff.linear1",
|
|
"ff.linear2",
|
|
"ff_context.linear1",
|
|
"ff_context.linear2"
|
|
],
|
|
"lora_rank": 4
|
|
},
|
|
"single_transformer_blocks": {
|
|
"block_range": {
|
|
"start": 17,
|
|
"end": 18
|
|
},
|
|
"layer_types": [
|
|
"proj_out",
|
|
"proj_mlp",
|
|
"attn.to_q",
|
|
"attn.to_k",
|
|
"attn.to_v"
|
|
],
|
|
"lora_rank": 4
|
|
}
|
|
},
|
|
"examples": {
|
|
"path": "../../../src/mflux/models/flux/variants/dreambooth/_example/images/",
|
|
"images": [
|
|
{
|
|
"image": "01.jpeg",
|
|
"prompt": "photo of sks dog"
|
|
},
|
|
{
|
|
"image": "02.jpeg",
|
|
"prompt": "photo of sks dog"
|
|
},
|
|
{
|
|
"image": "03.jpeg",
|
|
"prompt": "photo of sks dog"
|
|
},
|
|
{
|
|
"image": "04.jpeg",
|
|
"prompt": "photo of sks dog"
|
|
},
|
|
{
|
|
"image": "05.jpeg",
|
|
"prompt": "photo of sks dog"
|
|
}
|
|
]
|
|
}
|
|
} |