command: | |
- accelerate | |
- launch | |
- --config_file=/content/kohya-trainer/accelerate_config/config.yaml | |
- --num_cpu_threads_per_process=8 | |
- /content/kohya-trainer/train_network.py | |
- --output_name=kouhai | |
- --pretrained_model_name_or_path=/content/pretrained_model/anything-v3-fp32-pruned.safetensors | |
- --vae=/content/vae/anime.vae.pt | |
- --train_data_dir=/content/drive/MyDrive/kouhai | |
- --in_json=/content/drive/MyDrive/training_dir/meta_lat.json | |
- --output_dir=/content/drive/MyDrive/training_dir/output | |
- --network_dim=128 | |
- --network_alpha=128 | |
- --network_module=networks.lora | |
- --unet_lr=0.0001 | |
- --text_encoder_lr=5e-05 | |
- --optimizer_type=AdamW8bit | |
- --learning_rate=2e-06 | |
- --lr_scheduler=constant | |
- --lr_warmup_steps=250 | |
- --dataset_repeats=10 | |
- --resolution=512 | |
- --keep_tokens=1 | |
- --lowram | |
- --mixed_precision=fp16 | |
- --save_precision=fp16 | |
- --save_n_epoch_ratio=3 | |
- --save_model_as=safetensors | |
- --train_batch_size=4 | |
- --max_token_length=225 | |
- --max_train_steps=1500 | |
- --clip_skip=2 | |
- --logging_dir=/content/training_dir/logs | |
- --log_prefix=kouhai | |
- --shuffle_caption | |
- --xformers | |