Hugging Face's logo

MuYanchen
/
pokemon-lora

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files
xet
Metrics Community
MuYanchen commited on
Commit
377d7d0
·
1 Parent(s): 817ad59

End of training

Browse files
Files changed (25) hide show
  1. README.md +21 -0
  2. checkpoint-1000/optimizer.bin +3 -0
  3. checkpoint-1000/pytorch_model.bin +3 -0
  4. checkpoint-1000/random_states_0.pkl +3 -0
  5. checkpoint-1000/scaler.pt +3 -0
  6. checkpoint-1000/scheduler.bin +3 -0
  7. checkpoint-500/optimizer.bin +3 -0
  8. checkpoint-500/pytorch_model.bin +3 -0
  9. checkpoint-500/random_states_0.pkl +3 -0
  10. checkpoint-500/scaler.pt +3 -0
  11. checkpoint-500/scheduler.bin +3 -0
  12. image_0.png +0 -0
  13. image_1.png +0 -0
  14. image_2.png +0 -0
  15. image_3.png +0 -0
  16. logs/text2image-fine-tune/1682479528.7718108/events.out.tfevents.1682479528.amax.1895453.1 +3 -0
  17. logs/text2image-fine-tune/1682479528.7741604/hparams.yml +46 -0
  18. logs/text2image-fine-tune/1682479972.9220898/events.out.tfevents.1682479972.amax.1931988.1 +3 -0
  19. logs/text2image-fine-tune/1682479972.9244962/hparams.yml +46 -0
  20. logs/text2image-fine-tune/1682480152.87157/events.out.tfevents.1682480152.amax.1955083.1 +3 -0
  21. logs/text2image-fine-tune/1682480152.8739789/hparams.yml +46 -0
  22. logs/text2image-fine-tune/events.out.tfevents.1682479528.amax.1895453.0 +3 -0
  23. logs/text2image-fine-tune/events.out.tfevents.1682479972.amax.1931988.0 +3 -0
  24. logs/text2image-fine-tune/events.out.tfevents.1682480152.amax.1955083.0 +3 -0
  25. 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 - MuYanchen/pokemon-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-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc8f35de8b4317239e85691e32fa1eb5a7b142f91412865c21dab5001ae9cc93
3
+ size 6591621
checkpoint-1000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3eb29e22331948da0920b51452fa1416a43b48b8d9816f80f539cd262795ef33
3
+ size 3285965
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be78ae4fd4eea368d281b83b98a3ce5c04c969d3f3e4381cc88d21b18376d092
3
+ size 17747
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
3
+ size 557
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c6a1b58cc8b7c895d4a4ca2a65bf52b56826448cc3d15fabfc365819956c175
3
+ size 563
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:119cb8cb5c78b8ddff7b92d79250f4cce18d1abb8b2807bf81b478ecbe9329de
3
+ size 6591621
checkpoint-500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d49b5f1cec32fba08e8e7dea027dc62eacc7472bccba9ad067ba04130d32344
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:71ea307bfa5d6de388eb3fdc81041c8680d46e2809e38b4f3a3161d23e7d51d1
3
+ size 17747
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:96b42843a604e3681d1e20b27f085b13d50c8a8c4a629be0690985d351547c72
3
+ size 563
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/text2image-fine-tune/1682479528.7718108/events.out.tfevents.1682479528.amax.1895453.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a935860790f9de2a563b94f3c4ef70bdfdd473b8b87488fabcd716ae8692b6d
3
+ size 2117
logs/text2image-fine-tune/1682479528.7741604/hparams.yml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 556
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
34
+ push_to_hub: false
35
+ random_flip: true
36
+ report_to: tensorboard
37
+ resolution: 512
38
+ resume_from_checkpoint: null
39
+ revision: null
40
+ scale_lr: false
41
+ seed: null
42
+ train_batch_size: 1
43
+ train_data_dir: null
44
+ use_8bit_adam: false
45
+ validation_epochs: 1
46
+ validation_prompt: null
logs/text2image-fine-tune/1682479972.9220898/events.out.tfevents.1682479972.amax.1931988.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60877280eb1d28552ba91ae5ee70050596fce7e174ae0b387a390c425e6a26a5
3
+ size 2117
logs/text2image-fine-tune/1682479972.9244962/hparams.yml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
34
+ push_to_hub: false
35
+ random_flip: true
36
+ report_to: tensorboard
37
+ resolution: 512
38
+ resume_from_checkpoint: null
39
+ revision: null
40
+ scale_lr: false
41
+ seed: null
42
+ train_batch_size: 1
43
+ train_data_dir: null
44
+ use_8bit_adam: false
45
+ validation_epochs: 1
46
+ validation_prompt: null
logs/text2image-fine-tune/1682480152.87157/events.out.tfevents.1682480152.amax.1955083.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c44d946c98da340b285bd2db6f9992641e7c6ec17b28f3c4a2968eb209a7b159
3
+ size 2117
logs/text2image-fine-tune/1682480152.8739789/hparams.yml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 1000
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 19
31
+ num_validation_images: 4
32
+ output_dir: sd-model-finetuned-lora
33
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
34
+ push_to_hub: false
35
+ random_flip: true
36
+ report_to: tensorboard
37
+ resolution: 512
38
+ resume_from_checkpoint: null
39
+ revision: null
40
+ scale_lr: false
41
+ seed: null
42
+ train_batch_size: 1
43
+ train_data_dir: null
44
+ use_8bit_adam: false
45
+ validation_epochs: 1
46
+ validation_prompt: null
logs/text2image-fine-tune/events.out.tfevents.1682479528.amax.1895453.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06eaf0dbb629dc5c272504736ae0f33ab3bb948143fc700bfc7bc56b2be4e12d
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1682479972.amax.1931988.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35e571f6f303a5d4f4019bfac1a635efce045ea075d4d975838e0b3bce53931b
3
+ size 7115
logs/text2image-fine-tune/events.out.tfevents.1682480152.amax.1955083.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:761bfc1635df5305214852058840eea93111a4c2e34bdf0c737ec6e2c51227bf
3
+ size 48961
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b997c70dd3a02b60c487902436be9efffe45bf60c3d871e206d5b86c09dbba9
3
+ size 3287771