File size: 1,046 Bytes
01e31a2 |
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 |
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=Elysia
- --pretrained_model_name_or_path=/content/pretrained_model/model.ckpt
- --vae=/content/vae/anime.vae.pt
- --train_data_dir=/content/drive/MyDrive/elysua
- --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
- --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_epochs=10
- --clip_skip=2
- --logging_dir=/content/training_dir/logs
- --log_prefix=Elysia
- --shuffle_caption
- --xformers
|