Melonie commited on
Commit
8bde920
·
1 Parent(s): ad21b99

End of training

Browse files
Files changed (29) hide show
  1. README.md +21 -0
  2. checkpoint-500/optimizer.bin +3 -0
  3. checkpoint-500/pytorch_model.bin +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/text2image-fine-tune/1690223669.3093383/events.out.tfevents.1690223669.ventotene.70470.1 +3 -0
  12. logs/text2image-fine-tune/1690223669.3104954/hparams.yml +49 -0
  13. logs/text2image-fine-tune/1690224259.9579976/events.out.tfevents.1690224259.ventotene.90848.1 +3 -0
  14. logs/text2image-fine-tune/1690224259.9589646/hparams.yml +49 -0
  15. logs/text2image-fine-tune/1690224577.0793915/events.out.tfevents.1690224577.ventotene.93326.1 +3 -0
  16. logs/text2image-fine-tune/1690224577.080179/hparams.yml +49 -0
  17. logs/text2image-fine-tune/1690224772.319251/events.out.tfevents.1690224772.ventotene.96109.1 +3 -0
  18. logs/text2image-fine-tune/1690224772.3200748/hparams.yml +49 -0
  19. logs/text2image-fine-tune/1690224818.477918/events.out.tfevents.1690224818.ventotene.97433.1 +3 -0
  20. logs/text2image-fine-tune/1690224818.4787476/hparams.yml +49 -0
  21. logs/text2image-fine-tune/1690225118.2918434/events.out.tfevents.1690225118.ventotene.100376.1 +3 -0
  22. logs/text2image-fine-tune/1690225118.2926402/hparams.yml +49 -0
  23. logs/text2image-fine-tune/events.out.tfevents.1690223669.ventotene.70470.0 +3 -0
  24. logs/text2image-fine-tune/events.out.tfevents.1690224259.ventotene.90848.0 +3 -0
  25. logs/text2image-fine-tune/events.out.tfevents.1690224577.ventotene.93326.0 +3 -0
  26. logs/text2image-fine-tune/events.out.tfevents.1690224772.ventotene.96109.0 +3 -0
  27. logs/text2image-fine-tune/events.out.tfevents.1690224818.ventotene.97433.0 +3 -0
  28. logs/text2image-fine-tune/events.out.tfevents.1690225118.ventotene.100376.0 +3 -0
  29. pytorch_lora_weights.bin +3 -0
README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: runwayml/stable-diffusion-v1-5
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+
14
+ # LoRA text2image fine-tuning - Melonie/sd-model-finetuned-lora
15
+ These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the lambdalabs/pokemon-blip-captions dataset. You can find some example images in the following.
16
+
17
+ ![img_0](./image_0.png)
18
+ ![img_1](./image_1.png)
19
+ ![img_2](./image_2.png)
20
+ ![img_3](./image_3.png)
21
+
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:881849863804b18d294b0f122661cb58fc0001f017a44e4314055c1e6b5fc17c
3
+ size 6591685
checkpoint-500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5923424be854461a7ddf8ae7a8c2d43be59cf841491658a0b7ea6ef078948adf
3
+ size 3285965
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad947a6b95643ad4aaaa9174724d9e52e6e3494deb1b76fd166065cb07e791be
3
+ size 17683
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
3
+ size 557
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be9b855823954d49c1ce296b3f857a1c9801a9f569f7c1eeb228df6878a97ad0
3
+ size 563
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/text2image-fine-tune/1690223669.3093383/events.out.tfevents.1690223669.ventotene.70470.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:789a769ce00728e20624ea8879f481614970bb7db1bba17ef9a10d1b119fdacf
3
+ size 2146
logs/text2image-fine-tune/1690223669.3104954/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
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: null
29
+ noise_offset: 0
30
+ num_train_epochs: 284
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/1690224259.9579976/events.out.tfevents.1690224259.ventotene.90848.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3df02e59e639bb6e8348e9524d87a8c31cf88a23a0031ead7340b79c23e5a3e3
3
+ size 2146
logs/text2image-fine-tune/1690224259.9589646/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
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: 100
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 2
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/1690224577.0793915/events.out.tfevents.1690224577.ventotene.93326.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d874737d84dfed9ba9d191f042ba3e4d2035c866e6a163cca63daccf888f95e2
3
+ size 2146
logs/text2image-fine-tune/1690224577.080179/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
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: 100
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 2
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/1690224772.319251/events.out.tfevents.1690224772.ventotene.96109.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ac71ffeef36715b1e9d13d752966a6c8f7f360e03781963f18fdef8fb2f6f51
3
+ size 2148
logs/text2image-fine-tune/1690224772.3200748/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 500
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 5300
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 100
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/1690224818.477918/events.out.tfevents.1690224818.ventotene.97433.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e22c446f8b72e24477a08ce92511f6f71eb9623d02dd5de83e2bf6900369ef1
3
+ size 2148
logs/text2image-fine-tune/1690224818.4787476/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 500
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 5300
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 100
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/1690225118.2918434/events.out.tfevents.1690225118.ventotene.100376.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b08b3035b56fada0502d078800a4be59e921d20ec74af8e19447eb24d7057b6
3
+ size 2148
logs/text2image-fine-tune/1690225118.2926402/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 0.0001
21
+ local_rank: 0
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 500
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 5300
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 100
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ push_to_hub: false
36
+ random_flip: true
37
+ rank: 4
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
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: null
logs/text2image-fine-tune/events.out.tfevents.1690223669.ventotene.70470.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2fb75487837a0408a91f622353a4c5c9f2af4bb68ee6db2dbca2eabafc1e2ad
3
+ size 46070
logs/text2image-fine-tune/events.out.tfevents.1690224259.ventotene.90848.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6510926e71c648e3c5fb6ecd5967924df87646447eca9f85b23cc4923015616
3
+ size 4888
logs/text2image-fine-tune/events.out.tfevents.1690224577.ventotene.93326.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0ad4ba580386f02ad39570dbe181a841bbbc599a0791a69c4e1fe52a1fd89b2
3
+ size 4888
logs/text2image-fine-tune/events.out.tfevents.1690224772.ventotene.96109.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41e3843ed47dab0b76ffc2e8bd310f20529b50511ca4df4ea8d9867c3daaa5f1
3
+ size 1720
logs/text2image-fine-tune/events.out.tfevents.1690224818.ventotene.97433.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ba29b1ef6a08b671794ad124b0d3b84dfa94eb8205f5a14063efe3e84c3eff3
3
+ size 2920
logs/text2image-fine-tune/events.out.tfevents.1690225118.ventotene.100376.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b4e1c89ef114a08cfc8bb38eee16853b22522838c65b6cce5ae3e97135fc0c8
3
+ size 16719
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1070d96632caad3ae393721e3fb04f3404516df437cd0f5b6f5e6a699c250f4
3
+ size 3287771