File size: 1,705 Bytes
66df5ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
{
    "--resume_from_checkpoint": "latest",
    "--data_backend_config": "config/multidatabackend.json",
    "--aspect_bucket_rounding": 2,
    "--seed": 42,
    "--minimum_image_size": 0,
    "--disable_benchmark": false,
    "--output_dir": "output/models",
    "--lora_type": "lycoris",
    "--lycoris_config": "config/lycoris_config.json",
    "--max_train_steps": 10000,
    "--num_train_epochs": 0,
    "--checkpointing_steps": 100,
    "--checkpoints_total_limit": 5,
    "--hub_model_id": "sd35-spirited-away-lokr",
    "--push_to_hub": "true",
    "--push_checkpoints_to_hub": "true",
    "--model_card_safe_for_work": "true",
    "--tracker_project_name": "sd35-sa-lokr",
    "--tracker_run_name": "2",
    "--report_to": "wandb",
    "--model_type": "lora",
    "--pretrained_model_name_or_path": "stabilityai/stable-diffusion-3.5-large",
    "--model_family": "sd3",
    "--train_batch_size": 4,
    "--gradient_checkpointing": "true",
    "--caption_dropout_probability": 0.1,
    "--resolution_type": "pixel_area",
    "--resolution": 1024,
    "--validation_seed": 42,
    "--validation_steps": 100,
    "--validation_resolution": "1024x1024",
    "--validation_guidance": "4",
    "--validation_guidance_rescale": "0.0",
    "--validation_num_inference_steps": "20",
    "--validation_prompt": "A photo-realistic image of a cat",
    "--mixed_precision": "bf16",
    "--optimizer": "adamw_bf16",
    "--learning_rate": "1e-5",
    "--lr_scheduler": "constant_with_warmup",
    "--lr_warmup_steps": 100,
    "--validation_torch_compile": "true",
    "--max_grad_norm": 0.01,
    "--webhook_config": "config/sfw_webhook.json",
    "--user_prompt_library": "config/user_prompt_library.json"
}