k4d3 commited on
Commit
dc2fcd8
1 Parent(s): 68c65ea

Signed-off-by: Balazs Horvath <acsipont@gmail.com>

Files changed (4) hide show
  1. embeddings/EasyNegative.safetensors +3 -0
  2. tp_bs12x4 +68 -0
  3. tp_bs8x6 +68 -0
  4. train-pony.sh +14 -11
embeddings/EasyNegative.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c74b4e810b030f6b75fde959e2db678c268d07115b85356d3c0138ba5eb42340
3
+ size 24655
tp_bs12x4 ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #/usr/bin/env zsh
2
+
3
+ #"block_dims=1,1,1,1,1,1,1,1,1,1,1,1,8,8,8,8,8,8,8,8,8,8,8,8,8"
4
+ # alpha=1 @ dim=16 is the same lr than alpha=4 @ dim=256
5
+ cd ~/source/repos/sd-scripts &&
6
+ accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
7
+ --pretrained_model_name_or_path=/home/kade/ComfyUI/models/checkpoints/ponyDiffusionV6XL_v6StartWithThisOne.safetensors \
8
+ --train_data_dir=/home/kade/training_dir \
9
+ --resolution="1024,1024" \
10
+ --output_dir="/home/kade/output_dir" \
11
+ --output_name="chunie-v3" \
12
+ --enable_bucket \
13
+ --min_bucket_reso=256 \
14
+ --max_bucket_reso=2048 \
15
+ --network_alpha=4 \
16
+ --save_model_as="safetensors" \
17
+ --network_module="lycoris.kohya" \
18
+ --network_args \
19
+ "preset=full" \
20
+ "conv_dim=16" \
21
+ "conv_alpha=1" \
22
+ "rank_dropout=0" \
23
+ "module_dropout=0" \
24
+ "use_tucker=False" \
25
+ "use_scalar=False" \
26
+ "rank_dropout_scale=False" \
27
+ "algo=locon" \
28
+ "dora_wd=False" \
29
+ "train_norm=False" \
30
+ --network_dropout=0 \
31
+ --lr_scheduler="cosine" \
32
+ --lr_scheduler_args="num_cycles=0.375" \
33
+ --lr_warmup_steps=6 \
34
+ --learning_rate=0.0002 \
35
+ --unet_lr=0.0002 \
36
+ --text_encoder_lr=0.0001 \
37
+ --network_dim=8 \
38
+ --no_half_vae \
39
+ --flip_aug \
40
+ --save_every_n_steps=20 \
41
+ --mixed_precision="bf16" \
42
+ --save_precision="fp16" \
43
+ --cache_latents \
44
+ --cache_latents_to_disk \
45
+ --optimizer_type=ClybW \
46
+ --max_grad_norm=1 \
47
+ --max_data_loader_n_workers=8 \
48
+ --bucket_reso_steps=32 \
49
+ --multires_noise_iterations=12 \
50
+ --multires_noise_discount=0.4 \
51
+ --log_prefix=xl-locon \
52
+ --log_with=tensorboard \
53
+ --logging_dir=/home/kade/output_dir/logs \
54
+ --gradient_accumulation_steps=4 \
55
+ --gradient_checkpointing \
56
+ --train_batch_size=12 \
57
+ --dataset_repeats=1 \
58
+ --shuffle_caption \
59
+ --max_train_steps=500 \
60
+ --sdpa \
61
+ --sample_prompts=/home/kade/training_dir/sample-prompts.txt \
62
+ --sample_sampler="euler_a" \
63
+ --sample_every_n_steps=20 \
64
+ --caption_extension=".txt" \
65
+ --seed=1728871242 &&
66
+ cd ~
67
+
68
+ # --scale_weight_norms=1 \
tp_bs8x6 ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #/usr/bin/env zsh
2
+
3
+ #"block_dims=1,1,1,1,1,1,1,1,1,1,1,1,8,8,8,8,8,8,8,8,8,8,8,8,8"
4
+ # alpha=1 @ dim=16 is the same lr than alpha=4 @ dim=256
5
+ cd ~/source/repos/sd-scripts &&
6
+ accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
7
+ --pretrained_model_name_or_path=/home/kade/ComfyUI/models/checkpoints/ponyDiffusionV6XL_v6StartWithThisOne.safetensors \
8
+ --train_data_dir=/home/kade/training_dir \
9
+ --resolution="1024,1024" \
10
+ --output_dir="/home/kade/output_dir" \
11
+ --output_name="chunie-v3-bs8x12" \
12
+ --enable_bucket \
13
+ --min_bucket_reso=256 \
14
+ --max_bucket_reso=2048 \
15
+ --network_alpha=4 \
16
+ --save_model_as="safetensors" \
17
+ --network_module="lycoris.kohya" \
18
+ --network_args \
19
+ "preset=full" \
20
+ "conv_dim=16" \
21
+ "conv_alpha=1" \
22
+ "rank_dropout=0" \
23
+ "module_dropout=0" \
24
+ "use_tucker=False" \
25
+ "use_scalar=False" \
26
+ "rank_dropout_scale=False" \
27
+ "algo=locon" \
28
+ "dora_wd=False" \
29
+ "train_norm=False" \
30
+ --network_dropout=0 \
31
+ --lr_scheduler="cosine" \
32
+ --lr_scheduler_args="num_cycles=0.375" \
33
+ --lr_warmup_steps=6 \
34
+ --learning_rate=0.0002 \
35
+ --unet_lr=0.0002 \
36
+ --text_encoder_lr=0.0001 \
37
+ --network_dim=8 \
38
+ --no_half_vae \
39
+ --flip_aug \
40
+ --save_every_n_steps=20 \
41
+ --mixed_precision="bf16" \
42
+ --save_precision="fp16" \
43
+ --cache_latents \
44
+ --cache_latents_to_disk \
45
+ --optimizer_type=ClybW \
46
+ --max_grad_norm=1 \
47
+ --max_data_loader_n_workers=8 \
48
+ --bucket_reso_steps=32 \
49
+ --multires_noise_iterations=12 \
50
+ --multires_noise_discount=0.4 \
51
+ --log_prefix=xl-locon-bs8x12 \
52
+ --log_with=tensorboard \
53
+ --logging_dir=/home/kade/output_dir/logs \
54
+ --gradient_accumulation_steps=6 \
55
+ --gradient_checkpointing \
56
+ --train_batch_size=8 \
57
+ --dataset_repeats=1 \
58
+ --shuffle_caption \
59
+ --max_train_steps=500 \
60
+ --sdpa \
61
+ --sample_prompts=/home/kade/training_dir/sample-prompts.txt \
62
+ --sample_sampler="euler_a" \
63
+ --sample_every_n_steps=20 \
64
+ --caption_extension=".txt" \
65
+ --seed=1728871242 &&
66
+ cd ~
67
+
68
+ # --scale_weight_norms=1 \
train-pony.sh CHANGED
@@ -1,12 +1,14 @@
1
  #/usr/bin/env zsh
2
 
3
- cd ~/repos/sd-scripts &&
 
 
4
  accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
5
- --lowram \
6
  --pretrained_model_name_or_path=/home/kade/ComfyUI/models/checkpoints/ponyDiffusionV6XL_v6StartWithThisOne.safetensors \
7
  --train_data_dir=/home/kade/training_dir \
8
  --resolution="1024,1024" \
9
  --output_dir="/home/kade/output_dir" \
 
10
  --enable_bucket \
11
  --min_bucket_reso=256 \
12
  --max_bucket_reso=2048 \
@@ -15,8 +17,8 @@ accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
15
  --network_module="lycoris.kohya" \
16
  --network_args \
17
  "preset=full" \
18
- "conv_dim=256" \
19
- "conv_alpha=4" \
20
  "rank_dropout=0" \
21
  "module_dropout=0" \
22
  "use_tucker=False" \
@@ -27,14 +29,15 @@ accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
27
  "train_norm=False" \
28
  --network_dropout=0 \
29
  --lr_scheduler="cosine" \
 
 
30
  --learning_rate=0.0002 \
31
  --unet_lr=0.0002 \
32
  --text_encoder_lr=0.0001 \
33
  --network_dim=8 \
34
- --output_name=by_hax-v3 \
35
  --no_half_vae \
36
  --flip_aug \
37
- --save_every_n_steps=50 \
38
  --mixed_precision="bf16" \
39
  --save_precision="fp16" \
40
  --cache_latents \
@@ -48,16 +51,16 @@ accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
48
  --log_prefix=xl-locon \
49
  --log_with=tensorboard \
50
  --logging_dir=/home/kade/output_dir/logs \
51
- --gradient_accumulation_steps=6 \
52
  --gradient_checkpointing \
53
- --train_batch_size=8 \
54
- --dataset_repeats=0 \
55
  --shuffle_caption \
56
- --max_train_steps=800 \
57
  --sdpa \
58
  --sample_prompts=/home/kade/training_dir/sample-prompts.txt \
59
  --sample_sampler="euler_a" \
60
- --sample_every_n_steps=50 \
61
  --caption_extension=".txt" \
62
  --seed=1728871242 &&
63
  cd ~
 
1
  #/usr/bin/env zsh
2
 
3
+ #"block_dims=1,1,1,1,1,1,1,1,1,1,1,1,8,8,8,8,8,8,8,8,8,8,8,8,8"
4
+ # alpha=1 @ dim=16 is the same lr than alpha=4 @ dim=256
5
+ cd ~/source/repos/sd-scripts &&
6
  accelerate launch --num_cpu_threads_per_process=2 "./sdxl_train_network.py" \
 
7
  --pretrained_model_name_or_path=/home/kade/ComfyUI/models/checkpoints/ponyDiffusionV6XL_v6StartWithThisOne.safetensors \
8
  --train_data_dir=/home/kade/training_dir \
9
  --resolution="1024,1024" \
10
  --output_dir="/home/kade/output_dir" \
11
+ --output_name="chunie-v3" \
12
  --enable_bucket \
13
  --min_bucket_reso=256 \
14
  --max_bucket_reso=2048 \
 
17
  --network_module="lycoris.kohya" \
18
  --network_args \
19
  "preset=full" \
20
+ "conv_dim=16" \
21
+ "conv_alpha=1" \
22
  "rank_dropout=0" \
23
  "module_dropout=0" \
24
  "use_tucker=False" \
 
29
  "train_norm=False" \
30
  --network_dropout=0 \
31
  --lr_scheduler="cosine" \
32
+ --lr_scheduler_args="num_cycles=0.375" \
33
+ --lr_warmup_steps=6 \
34
  --learning_rate=0.0002 \
35
  --unet_lr=0.0002 \
36
  --text_encoder_lr=0.0001 \
37
  --network_dim=8 \
 
38
  --no_half_vae \
39
  --flip_aug \
40
+ --save_every_n_steps=20 \
41
  --mixed_precision="bf16" \
42
  --save_precision="fp16" \
43
  --cache_latents \
 
51
  --log_prefix=xl-locon \
52
  --log_with=tensorboard \
53
  --logging_dir=/home/kade/output_dir/logs \
54
+ --gradient_accumulation_steps=4 \
55
  --gradient_checkpointing \
56
+ --train_batch_size=12 \
57
+ --dataset_repeats=1 \
58
  --shuffle_caption \
59
+ --max_train_steps=500 \
60
  --sdpa \
61
  --sample_prompts=/home/kade/training_dir/sample-prompts.txt \
62
  --sample_sampler="euler_a" \
63
+ --sample_every_n_steps=20 \
64
  --caption_extension=".txt" \
65
  --seed=1728871242 &&
66
  cd ~