Jayw1 commited on
Commit
dc2e178
·
verified ·
1 Parent(s): 37a7d19

End of training

Browse files
Files changed (23) hide show
  1. README.md +48 -0
  2. checkpoint-500/optimizer.bin +3 -0
  3. checkpoint-500/pytorch_lora_weights.safetensors +3 -0
  4. checkpoint-500/random_states_0.pkl +3 -0
  5. checkpoint-500/scaler.pt +3 -0
  6. checkpoint-500/scheduler.bin +3 -0
  7. image_0.png +0 -0
  8. image_1.png +0 -0
  9. image_2.png +0 -0
  10. image_3.png +0 -0
  11. logs/dreambooth-lora-playground/1729158712.5517864/events.out.tfevents.1729158712.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2438567.1 +3 -0
  12. logs/dreambooth-lora-playground/1729158712.5590553/hparams.yml +70 -0
  13. logs/dreambooth-lora-playground/1729166423.3885906/events.out.tfevents.1729166423.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2538484.1 +3 -0
  14. logs/dreambooth-lora-playground/1729166423.404417/hparams.yml +70 -0
  15. logs/dreambooth-lora-playground/1729167872.7762794/events.out.tfevents.1729167872.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2564485.1 +3 -0
  16. logs/dreambooth-lora-playground/1729167872.7831347/hparams.yml +70 -0
  17. logs/dreambooth-lora-playground/1729168513.077605/events.out.tfevents.1729168513.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2583366.1 +3 -0
  18. logs/dreambooth-lora-playground/1729168513.0846903/hparams.yml +70 -0
  19. logs/dreambooth-lora-playground/events.out.tfevents.1729158712.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2438567.0 +3 -0
  20. logs/dreambooth-lora-playground/events.out.tfevents.1729166423.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2538484.0 +3 -0
  21. logs/dreambooth-lora-playground/events.out.tfevents.1729167872.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2564485.0 +3 -0
  22. logs/dreambooth-lora-playground/events.out.tfevents.1729168513.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2583366.0 +3 -0
  23. pytorch_lora_weights.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: playgroundai/playground-v2.5-1024px-aesthetic
3
+ library_name: diffusers
4
+ license: creativeml-openrail-m
5
+ tags:
6
+ - stable-diffusion-xl
7
+ - stable-diffusion-xl-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - diffusers-training
11
+ - lora
12
+ inference: true
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+
19
+ # LoRA text2image fine-tuning - Jayw1/lora_baimiao
20
+
21
+ These are LoRA adaption weights for playgroundai/playground-v2.5-1024px-aesthetic. The weights were fine-tuned on the /mnt/afs/data/hjw/diffusers/baimiao dataset. You can find some example images in the following.
22
+
23
+ ![img_0](./image_0.png)
24
+ ![img_1](./image_1.png)
25
+ ![img_2](./image_2.png)
26
+ ![img_3](./image_3.png)
27
+
28
+
29
+ LoRA for the text encoder was enabled: False.
30
+
31
+ Special VAE used for training: None.
32
+
33
+
34
+ ## Intended uses & limitations
35
+
36
+ #### How to use
37
+
38
+ ```python
39
+ # TODO: add an example code snippet for running this diffusion pipeline
40
+ ```
41
+
42
+ #### Limitations and bias
43
+
44
+ [TODO: provide examples of latent issues and potential remediations]
45
+
46
+ ## Training details
47
+
48
+ [TODO: describe the data used to train the model]
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0632df4333808112b132c3200a9fd4da9e47bf6750885fb519564fa046867b1c
3
+ size 15134460
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38007045909dc93723b4a16be0974aa7350447932e15c52721e98bad7b73d0ed
3
+ size 23390424
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f6d2a0fd02ef33fd9271bdcbddee2da0c21ecc8e7a5c1a17935c55112c19c2b
3
+ size 14408
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18b984273ea2d45b7ffb1d047bb359d93111e41fcad70d16a1b453fd38f72636
3
+ size 988
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c
3
+ size 1000
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/dreambooth-lora-playground/1729158712.5517864/events.out.tfevents.1729158712.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2438567.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f60f388badf416dc4c0eea9950b8b1d211d6710821e33caf0ce8e34abf9406c0
3
+ size 3230
logs/dreambooth-lora-playground/1729158712.5590553/hparams.yml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: text
9
+ center_crop: false
10
+ checkpointing_steps: 500
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: /mnt/afs/data/hjw/diffusers/baimiao
17
+ do_edm_style_training: true
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: null
25
+ instance_prompt: xubeihong22
26
+ learning_rate: 0.0001
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 500
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 50
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: lora_baimiao
41
+ output_kohya_format: false
42
+ pretrained_model_name_or_path: playgroundai/playground-v2.5-1024px-aesthetic
43
+ pretrained_vae_model_name_or_path: null
44
+ prior_generation_precision: null
45
+ prior_loss_weight: 1.0
46
+ prodigy_beta3: null
47
+ prodigy_decouple: true
48
+ prodigy_safeguard_warmup: true
49
+ prodigy_use_bias_correction: true
50
+ push_to_hub: true
51
+ random_flip: false
52
+ rank: 4
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 256
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 222
61
+ snr_gamma: null
62
+ text_encoder_lr: 5.0e-06
63
+ train_batch_size: 1
64
+ train_text_encoder: false
65
+ use_8bit_adam: true
66
+ use_dora: false
67
+ validation_epochs: 40
68
+ validation_prompt: xubeihong
69
+ variant: null
70
+ with_prior_preservation: false
logs/dreambooth-lora-playground/1729166423.3885906/events.out.tfevents.1729166423.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2538484.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f946e21099856bceaae0bcfdcc30651a0c658410e769c0f37e389cf52522084e
3
+ size 3228
logs/dreambooth-lora-playground/1729166423.404417/hparams.yml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: text
9
+ center_crop: false
10
+ checkpointing_steps: 500
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: /mnt/afs/data/hjw/diffusers/baimiao
17
+ do_edm_style_training: true
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: null
25
+ instance_prompt: xubeihong22
26
+ learning_rate: 0.0001
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 500
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 50
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: lora_baimiao
41
+ output_kohya_format: false
42
+ pretrained_model_name_or_path: playgroundai/playground-v2.5-1024px-aesthetic
43
+ pretrained_vae_model_name_or_path: null
44
+ prior_generation_precision: null
45
+ prior_loss_weight: 1.0
46
+ prodigy_beta3: null
47
+ prodigy_decouple: true
48
+ prodigy_safeguard_warmup: true
49
+ prodigy_use_bias_correction: true
50
+ push_to_hub: true
51
+ random_flip: false
52
+ rank: 4
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 256
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 222
61
+ snr_gamma: null
62
+ text_encoder_lr: 5.0e-06
63
+ train_batch_size: 1
64
+ train_text_encoder: false
65
+ use_8bit_adam: true
66
+ use_dora: false
67
+ validation_epochs: 40
68
+ validation_prompt: baimiao
69
+ variant: null
70
+ with_prior_preservation: false
logs/dreambooth-lora-playground/1729167872.7762794/events.out.tfevents.1729167872.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2564485.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:629c912f0726373aeff6c11eb97be72c8e7887eb7e1aca916f86a550e1186ded
3
+ size 3228
logs/dreambooth-lora-playground/1729167872.7831347/hparams.yml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: text
9
+ center_crop: false
10
+ checkpointing_steps: 500
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: /mnt/afs/data/hjw/diffusers/baimiao
17
+ do_edm_style_training: true
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: null
25
+ instance_prompt: xubeihong22
26
+ learning_rate: 0.0001
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 500
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 50
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: lora_baimiao
41
+ output_kohya_format: false
42
+ pretrained_model_name_or_path: playgroundai/playground-v2.5-1024px-aesthetic
43
+ pretrained_vae_model_name_or_path: null
44
+ prior_generation_precision: null
45
+ prior_loss_weight: 1.0
46
+ prodigy_beta3: null
47
+ prodigy_decouple: true
48
+ prodigy_safeguard_warmup: true
49
+ prodigy_use_bias_correction: true
50
+ push_to_hub: true
51
+ random_flip: false
52
+ rank: 4
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 256
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 222
61
+ snr_gamma: null
62
+ text_encoder_lr: 5.0e-06
63
+ train_batch_size: 1
64
+ train_text_encoder: false
65
+ use_8bit_adam: true
66
+ use_dora: false
67
+ validation_epochs: 40
68
+ validation_prompt: baimiao
69
+ variant: null
70
+ with_prior_preservation: false
logs/dreambooth-lora-playground/1729168513.077605/events.out.tfevents.1729168513.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2583366.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a523a7a5ddcee95d615acceb2674f5e6cb84cd7ef16715ec129f9f55c7d9d5f9
3
+ size 3228
logs/dreambooth-lora-playground/1729168513.0846903/hparams.yml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: text
9
+ center_crop: false
10
+ checkpointing_steps: 500
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: /mnt/afs/data/hjw/diffusers/baimiao
17
+ do_edm_style_training: true
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: null
25
+ instance_prompt: xubeihong22
26
+ learning_rate: 0.0001
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 500
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 50
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: lora_baimiao
41
+ output_kohya_format: false
42
+ pretrained_model_name_or_path: playgroundai/playground-v2.5-1024px-aesthetic
43
+ pretrained_vae_model_name_or_path: null
44
+ prior_generation_precision: null
45
+ prior_loss_weight: 1.0
46
+ prodigy_beta3: null
47
+ prodigy_decouple: true
48
+ prodigy_safeguard_warmup: true
49
+ prodigy_use_bias_correction: true
50
+ push_to_hub: true
51
+ random_flip: false
52
+ rank: 4
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 256
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 222
61
+ snr_gamma: null
62
+ text_encoder_lr: 5.0e-06
63
+ train_batch_size: 1
64
+ train_text_encoder: false
65
+ use_8bit_adam: true
66
+ use_dora: false
67
+ validation_epochs: 40
68
+ validation_prompt: baimiao
69
+ variant: null
70
+ with_prior_preservation: false
logs/dreambooth-lora-playground/events.out.tfevents.1729158712.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2438567.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e32d4c6969a213c1812bc8ea9667c747c9b93201e1888279756a8543cdc8aec
3
+ size 6736983
logs/dreambooth-lora-playground/events.out.tfevents.1729166423.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2538484.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:faf366e3218de89d59e9078215c5c06b46f5be479d7f81d87612f10fa12e6038
3
+ size 7363987
logs/dreambooth-lora-playground/events.out.tfevents.1729167872.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2564485.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81fa4f0f29b0a07e6cb5191260588bcd93b8595c1d3a7608f9fc3f27e7a8d32f
3
+ size 7363987
logs/dreambooth-lora-playground/events.out.tfevents.1729168513.3ba3897f-4fac-11ef-902f-3e6491b2e6f5.2583366.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98a4b39fd6051c4a17dec0f30b0bc5e39e0a6fa02d5a4c248dc82bdb6c9812d2
3
+ size 7363987
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5296c5ba66702726d9c5523d3f26dcb1c882d1a61eaae280a4f746af31d7b368
3
+ size 23390424