| { | |
| "--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/peft_config.1.4b.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": "8", | |
| "--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": "5", | |
| "--validation_guidance_rescale": "0.0", | |
| "--validation_num_inference_steps": "20", | |
| "--validation_prompt": "A photo-realistic image of a cat", | |
| "--validation_negative_prompt": "blurry, cropped, ugly", | |
| "--mixed_precision": "bf16", | |
| "--optimizer": "adamw_bf16", | |
| "--learning_rate": "1e-7", | |
| "--lr_scheduler": "constant_with_warmup", | |
| "--lr_warmup_steps": 100, | |
| "--base_model_precision": "int8-quanto", | |
| "--validation_torch_compile": "true", | |
| "--max_grad_norm": 0.01, | |
| "--init_lokr_norm": "1e-3", | |
| "--webhook_config": "config/sfw_webhook.json", | |
| "--user_prompt_library": "config/user_prompt_library.json" | |
| } |