yifei28 commited on
Commit
99d20a8
1 Parent(s): 0e361c4

End of training

Browse files
Files changed (42) hide show
  1. README.md +6 -0
  2. checkpoint-12000/model.safetensors +3 -0
  3. checkpoint-12000/optimizer.bin +3 -0
  4. checkpoint-12000/pytorch_lora_weights.safetensors +3 -0
  5. checkpoint-12000/random_states_0.pkl +3 -0
  6. checkpoint-12000/scaler.pt +3 -0
  7. checkpoint-12000/scheduler.bin +3 -0
  8. checkpoint-15000/model.safetensors +3 -0
  9. checkpoint-15000/optimizer.bin +3 -0
  10. checkpoint-15000/pytorch_lora_weights.safetensors +3 -0
  11. checkpoint-15000/random_states_0.pkl +3 -0
  12. checkpoint-15000/scaler.pt +3 -0
  13. checkpoint-15000/scheduler.bin +3 -0
  14. checkpoint-3000/model.safetensors +3 -0
  15. checkpoint-3000/optimizer.bin +3 -0
  16. checkpoint-3000/pytorch_lora_weights.safetensors +3 -0
  17. checkpoint-3000/random_states_0.pkl +3 -0
  18. checkpoint-3000/scaler.pt +3 -0
  19. checkpoint-3000/scheduler.bin +3 -0
  20. checkpoint-6000/model.safetensors +3 -0
  21. checkpoint-6000/optimizer.bin +3 -0
  22. checkpoint-6000/pytorch_lora_weights.safetensors +3 -0
  23. checkpoint-6000/random_states_0.pkl +3 -0
  24. checkpoint-6000/scaler.pt +3 -0
  25. checkpoint-6000/scheduler.bin +3 -0
  26. checkpoint-9000/model.safetensors +3 -0
  27. checkpoint-9000/optimizer.bin +3 -0
  28. checkpoint-9000/pytorch_lora_weights.safetensors +3 -0
  29. checkpoint-9000/random_states_0.pkl +3 -0
  30. checkpoint-9000/scaler.pt +3 -0
  31. checkpoint-9000/scheduler.bin +3 -0
  32. image_0.png +0 -0
  33. image_1.png +0 -0
  34. image_2.png +0 -0
  35. image_3.png +0 -0
  36. logs/text2image-fine-tune/1720813367.3979056/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.1 +3 -0
  37. logs/text2image-fine-tune/1720813367.399584/hparams.yml +50 -0
  38. logs/text2image-fine-tune/1720834818.385705/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.1 +3 -0
  39. logs/text2image-fine-tune/1720834818.387928/hparams.yml +50 -0
  40. logs/text2image-fine-tune/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.0 +3 -0
  41. logs/text2image-fine-tune/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.0 +3 -0
  42. pytorch_lora_weights.safetensors +1 -1
README.md CHANGED
@@ -9,6 +9,12 @@ tags:
9
  - diffusers
10
  - diffusers-training
11
  - lora
 
 
 
 
 
 
12
  inference: true
13
  ---
14
 
 
9
  - diffusers
10
  - diffusers-training
11
  - lora
12
+ - stable-diffusion
13
+ - stable-diffusion-diffusers
14
+ - text-to-image
15
+ - diffusers
16
+ - diffusers-training
17
+ - lora
18
  inference: true
19
  ---
20
 
checkpoint-12000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f8fb2e045ecbb00b7ac6e79dd10643ed98b644fe0f5abd61ed4909a64c0a203
3
+ size 1722353544
checkpoint-12000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25c2ded06af07056dbfb529f3cb9d18fdfaa701e9d0c39810a5e750f98cf3692
3
+ size 6592122
checkpoint-12000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:253d906c3199e0022103246f9fb5c119fa8b8a51b0015f495b8c35ad2fc1a15b
3
+ size 3226184
checkpoint-12000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a65a93f37b9bf6197d180eb449d5cbbfe678820d6c038e86e1bad931d8ab1ee9
3
+ size 14408
checkpoint-12000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d1f9d6a63312bf6188d8844a7d621f2ecda20b6c0aee9e19ddcea1c480954ba
3
+ size 988
checkpoint-12000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9921b7da5c99c86f03a3f19361ffe4ae9a1d7472d81b1adc17eaae9fcbef933b
3
+ size 1000
checkpoint-15000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d57f7fec10a0efa182da5fe3269647bca984faeaa2b88683ce889793f934c843
3
+ size 1722353544
checkpoint-15000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9080f48edfd4f9a1068eb02f3c2148e08dce5f29307e87a2da1a4f9a729f6a18
3
+ size 6592122
checkpoint-15000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2780d5e3b74c027ab40240c1ee54adc178cb3f82929636a098e0db4e20f26f3b
3
+ size 3226184
checkpoint-15000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d483c2bfb80aa778afd8aac3efabc5af3e0f72f9f667f8078925883b4a2ee4b
3
+ size 14408
checkpoint-15000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80f2829dbe9102c995677cfbb16e04c7f863ba8bbe9c0f80b5f219d485af0321
3
+ size 988
checkpoint-15000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25ac02ce31bea574afc7001488812dcd64ad3bff8601eb47775c323133594325
3
+ size 1000
checkpoint-3000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a12cd2ca8bfc6b097dfa3b05ce2ffe02266061ea1a8cc60c94576ac1b7914f25
3
+ size 1722353544
checkpoint-3000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d61d8bcfe62ea188ab24dc67aab56a1d6de3c1a69dc5fc51fe3d09dbbc1872d
3
+ size 6592122
checkpoint-3000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4b25c6be6862e5a8ee3d70665352a971668e3d3d464e6e79baa2247e358a797
3
+ size 3226184
checkpoint-3000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ddeaab2465529c41de91e04719cc693587c453d105f4056932d547087ed36432
3
+ size 14408
checkpoint-3000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21aba8ed0f38ed1c04994c10a9ca7e9925e55ef2ed51283c43ff8e2cce78585f
3
+ size 988
checkpoint-3000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89df8bcca59003290e3a854d22c2892d49d485440014cbb479fc1d4c852903d2
3
+ size 1000
checkpoint-6000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce463b97700e2d92b996038e6457ef55e474b65cb27e516370c5ee91158bcd6d
3
+ size 1722353544
checkpoint-6000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f97717f4f5ae0f4229357164ef674454e3bd77d8c62e55b1f0b62e197c48eed
3
+ size 6592122
checkpoint-6000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d0621810a41812125f73d4909fe5b7680b4ad5a81397a53c6408be8ed86fcc6
3
+ size 3226184
checkpoint-6000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc9de4c60e5996f0ac0269285f6661eb9f2a316e495f3559829e24e7d532424c
3
+ size 14408
checkpoint-6000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23c6abec96b9d238cb94e3e6ca47ebac7366e65b2c1f7f780949153075612b6e
3
+ size 988
checkpoint-6000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8efbe3d12cb8a569b24bc75287fb9269526a43518a997085b3fd06b850026b3
3
+ size 1000
checkpoint-9000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5e5742c22e3f66aa24f185a01923928ee22f5745d331fb7bea07f64cebe9323
3
+ size 1722353544
checkpoint-9000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d62607d1e13bfd7cbe199917375c4daab52ee3b022ddbd8ce377ce821c1f422
3
+ size 6592122
checkpoint-9000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:983d2fb68429c85adcdb097d188c59d3a92a0e86f8a8ee59cf5be5fc259042c0
3
+ size 3226184
checkpoint-9000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dda0f9b951447daef87560116be5b89f5a9c844e9dcd2c49a9cfb6b8261abb3e
3
+ size 14408
checkpoint-9000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ad6c50c35b53713927f28732052bf0f3bf5da247a85776559de9817d2d278c7
3
+ size 988
checkpoint-9000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b79c3422e1af76ea696e5b4dc00cd35dc6ee89af188fa2a23482f281e12789d
3
+ size 1000
image_0.png CHANGED
image_1.png CHANGED
image_2.png CHANGED
image_3.png CHANGED
logs/text2image-fine-tune/1720813367.3979056/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e76f06865c30b01cd00d896d91e22254df56a09abf1060ae1761bdc9b38aea58
3
+ size 2357
logs/text2image-fine-tune/1720813367.399584/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 3000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/naruto-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_scheduler: cosine
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 15000
28
+ mixed_precision: fp16
29
+ noise_offset: 0
30
+ num_train_epochs: 50
31
+ num_validation_images: 4
32
+ output_dir: sd-naruto-model
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: true
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: latest
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 1337
44
+ snr_gamma: null
45
+ train_batch_size: 1
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: Stephen Curry with naruto style
50
+ variant: null
logs/text2image-fine-tune/1720834818.385705/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f0dc22191fef9e949e3e7c82927e7c46e19100eb21cccde5a76d7d0fec8719f
3
+ size 2357
logs/text2image-fine-tune/1720834818.387928/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 3000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/naruto-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_scheduler: cosine
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 15000
28
+ mixed_precision: fp16
29
+ noise_offset: 0
30
+ num_train_epochs: 50
31
+ num_validation_images: 4
32
+ output_dir: sd-naruto-model
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: true
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: latest
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 1337
44
+ snr_gamma: null
45
+ train_batch_size: 1
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: Stephen Curry with Naruto style
50
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfb1c7835dcabc1424d5e3949029ee28d636714acb5053932f397e044094cc62
3
+ size 72467702
logs/text2image-fine-tune/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26cd859ab7a1719c071635bc00533a9f890a80e83298d76c33c6edb5aba02220
3
+ size 20398868
pytorch_lora_weights.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aa8472dc2ffdb17637f8d254f4aac4834d8cf8e36f37a95ee9fa6c3af728c547
3
  size 3226184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2780d5e3b74c027ab40240c1ee54adc178cb3f82929636a098e0db4e20f26f3b
3
  size 3226184