Hugging Face's logo

Alc141
/
Poke_txt2img

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files Metrics Community
Alc141 commited on
Commit
d6d0c67
·
1 Parent(s): 4b81928

End of training

Browse files
Files changed (33) hide show
  1. README.md +21 -0
  2. checkpoint-10000/optimizer.bin +3 -0
  3. checkpoint-10000/pytorch_model.bin +3 -0
  4. checkpoint-10000/random_states_0.pkl +3 -0
  5. checkpoint-10000/scaler.pt +3 -0
  6. checkpoint-10000/scheduler.bin +3 -0
  7. checkpoint-15000/optimizer.bin +3 -0
  8. checkpoint-15000/pytorch_model.bin +3 -0
  9. checkpoint-15000/random_states_0.pkl +3 -0
  10. checkpoint-15000/scaler.pt +3 -0
  11. checkpoint-15000/scheduler.bin +3 -0
  12. checkpoint-5000/optimizer.bin +3 -0
  13. checkpoint-5000/pytorch_model.bin +3 -0
  14. checkpoint-5000/random_states_0.pkl +3 -0
  15. checkpoint-5000/scaler.pt +3 -0
  16. checkpoint-5000/scheduler.bin +3 -0
  17. image_0.png +0 -0
  18. image_1.png +0 -0
  19. image_2.png +0 -0
  20. image_3.png +0 -0
  21. logs/text2image-fine-tune/1700304079.5078897/events.out.tfevents.1700304079.fecdf3853112.1752.1 +3 -0
  22. logs/text2image-fine-tune/1700304079.5095584/hparams.yml +49 -0
  23. logs/text2image-fine-tune/1700304353.6271625/events.out.tfevents.1700304353.fecdf3853112.3092.1 +3 -0
  24. logs/text2image-fine-tune/1700304353.6288896/hparams.yml +49 -0
  25. logs/text2image-fine-tune/1700304672.7178388/events.out.tfevents.1700304672.fecdf3853112.4630.1 +3 -0
  26. logs/text2image-fine-tune/1700304672.7211475/hparams.yml +49 -0
  27. logs/text2image-fine-tune/1700304744.4779136/events.out.tfevents.1700304744.fecdf3853112.4926.1 +3 -0
  28. logs/text2image-fine-tune/1700304744.4797087/hparams.yml +49 -0
  29. logs/text2image-fine-tune/events.out.tfevents.1700304079.fecdf3853112.1752.0 +3 -0
  30. logs/text2image-fine-tune/events.out.tfevents.1700304353.fecdf3853112.3092.0 +3 -0
  31. logs/text2image-fine-tune/events.out.tfevents.1700304672.fecdf3853112.4630.0 +3 -0
  32. logs/text2image-fine-tune/events.out.tfevents.1700304744.fecdf3853112.4926.0 +3 -0
  33. pytorch_lora_weights.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: CompVis/stable-diffusion-v1-4
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 - Alc141/Poke_txt2img
15
+ These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. 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-10000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2430fc9cb089360940ce3a79b039ed18fb1511ecdddb67e217d22a2716c2f630
3
+ size 6592122
checkpoint-10000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4dcf4b7f3244e7f11d77523b1306fb6cfa8aff512192573fe38c2d3af5111bf3
3
+ size 3286410
checkpoint-10000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66fc27c28ba4ff44f0f7f9e09a629dfefb8333a9e231b852703158dc420a29af
3
+ size 14344
checkpoint-10000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d49367aa0f19981d0484e3ef5c1b009e6bdd8e5c0d8ef08f2f8d235b9f843816
3
+ size 988
checkpoint-10000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2bd2fea55575e9e5201cd70f190230d22db4f67ec868e253d7e182dd49a19ee
3
+ size 1000
checkpoint-15000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d707d48610a8a9bd3c5367441e211afce5ea149460e77d9b460c1886165a5cbf
3
+ size 6592122
checkpoint-15000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61c7497f602b0469faf655cabbf84c936e77fab94f5fcac07f225ed13427d7ae
3
+ size 3286410
checkpoint-15000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4612b49ad676504bf1c752e6b2a7753008629852501c6b16769e7b6b5dfcd4cc
3
+ size 14344
checkpoint-15000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74db1924c929375827a2590439c1f65ed9ff2e5faca8c9e28a5b41fc8ee57fd6
3
+ size 988
checkpoint-15000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d1c83e8d876afd205935f688a36b62a8852dd54cc9e39988b330fa4db556687
3
+ size 1000
checkpoint-5000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4caee8894da38ebefaf66ac94709a0de2848253378fc8bc0ebe967d4906dfbd0
3
+ size 6592122
checkpoint-5000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc11038a10596ce3246fe52b96f6e2682c7bdf34d5d16fbf62143399a0cae2af
3
+ size 3286410
checkpoint-5000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09a2fe084b9934108ddf151b91bb54061bd60857b5962c821b96804899040beb
3
+ size 14344
checkpoint-5000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41216328ab75de937007afb04d76156949bedb908461001a451c8991c2ba8cca
3
+ size 988
checkpoint-5000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2734d36d08357236113a0d0e82c9a606fbb22a775c8bb9049c19d997ef37430
3
+ size 1000
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/text2image-fine-tune/1700304079.5078897/events.out.tfevents.1700304079.fecdf3853112.1752.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b6cacfe93788948ac7a4718ffe4fcb65f551f1c0c0758a9f0b350773de01011
3
+ size 2233
logs/text2image-fine-tune/1700304079.5095584/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: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
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: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 83300
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 100
31
+ num_validation_images: 4
32
+ output_dir: Poke_txt2img
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
35
+ push_to_hub: true
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: 42
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: cute dragon creature
logs/text2image-fine-tune/1700304353.6271625/events.out.tfevents.1700304353.fecdf3853112.3092.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9bf5f006061d9c4af35e5800606bfaafd841bbc238cca8670edc2af479d4d19
3
+ size 2233
logs/text2image-fine-tune/1700304353.6288896/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: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
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: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 2780
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 20
31
+ num_validation_images: 4
32
+ output_dir: Poke_txt2img
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
35
+ push_to_hub: true
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: 42
44
+ snr_gamma: null
45
+ train_batch_size: 6
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: cute dragon creature
logs/text2image-fine-tune/1700304672.7178388/events.out.tfevents.1700304672.fecdf3853112.4630.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:317f3272d5ccf4b205bc4d060900e1fd1dfe8a545a93639e242b486076ee44df
3
+ size 2233
logs/text2image-fine-tune/1700304672.7211475/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: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
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: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 2780
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 20
31
+ num_validation_images: 4
32
+ output_dir: Poke_txt2img
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
35
+ push_to_hub: true
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: 42
44
+ snr_gamma: null
45
+ train_batch_size: 6
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ validation_epochs: 1
49
+ validation_prompt: cute dragon creature
logs/text2image-fine-tune/1700304744.4779136/events.out.tfevents.1700304744.fecdf3853112.4926.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02e108510c5752892c13cf362a8ceb09a968b09f09231546ab8ca9cfa1ee20d7
3
+ size 2233
logs/text2image-fine-tune/1700304744.4797087/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: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: lambdalabs/pokemon-blip-captions
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
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: -1
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_samples: null
27
+ max_train_steps: 16660
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 20
31
+ num_validation_images: 4
32
+ output_dir: Poke_txt2img
33
+ prediction_type: null
34
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
35
+ push_to_hub: true
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: 42
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: cute dragon creature
logs/text2image-fine-tune/events.out.tfevents.1700304079.fecdf3853112.1752.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29feec55bc7a95d9442ebc57b7a61701dd5f6d041fc843a52b65fb92629e6b2e
3
+ size 26029
logs/text2image-fine-tune/events.out.tfevents.1700304353.fecdf3853112.3092.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57cda6b485a407dbc7c6909d13a91f1dbee6057c076de8c61de38a47e7080b1c
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1700304672.fecdf3853112.4630.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e28b146781754d5b0459dd899729069af55ae25219a1254561a6d60cce30cafc
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1700304744.fecdf3853112.4926.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60dc5f4625e742cfe795a61f3bd98da76668d984d92c60c1e5f449f5cb4270c2
3
+ size 25841477
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1774d6d731b10efa6a550fde7ac3e486c508fb9991f356f0d393c4cbbde0a037
3
+ size 3227336