Hugging Face's logo

ButterChicken98
/
plantVillage-stableDiffusion

Text-to-Image
Diffusers
TensorBoard
Safetensors
stable-diffusion
stable-diffusion-diffusers
diffusers-training
lora
Model card Files Metrics Community
ButterChicken98 commited on
Commit
006c01f
·
verified ·
1 Parent(s): 096fef2

End of training

Browse files
Files changed (42) hide show
  1. README.md +43 -0
  2. checkpoint-1000/model.safetensors +3 -0
  3. checkpoint-1000/optimizer.bin +3 -0
  4. checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
  5. checkpoint-1000/random_states_0.pkl +3 -0
  6. checkpoint-1000/scheduler.bin +3 -0
  7. checkpoint-1500/model.safetensors +3 -0
  8. checkpoint-1500/optimizer.bin +3 -0
  9. checkpoint-1500/pytorch_lora_weights.safetensors +3 -0
  10. checkpoint-1500/random_states_0.pkl +3 -0
  11. checkpoint-1500/scheduler.bin +3 -0
  12. checkpoint-2000/model.safetensors +3 -0
  13. checkpoint-2000/optimizer.bin +3 -0
  14. checkpoint-2000/pytorch_lora_weights.safetensors +3 -0
  15. checkpoint-2000/random_states_0.pkl +3 -0
  16. checkpoint-2000/scheduler.bin +3 -0
  17. checkpoint-2500/model.safetensors +3 -0
  18. checkpoint-2500/optimizer.bin +3 -0
  19. checkpoint-2500/pytorch_lora_weights.safetensors +3 -0
  20. checkpoint-2500/random_states_0.pkl +3 -0
  21. checkpoint-2500/scheduler.bin +3 -0
  22. checkpoint-3000/model.safetensors +3 -0
  23. checkpoint-3000/optimizer.bin +3 -0
  24. checkpoint-3000/pytorch_lora_weights.safetensors +3 -0
  25. checkpoint-3000/random_states_0.pkl +3 -0
  26. checkpoint-3000/scheduler.bin +3 -0
  27. checkpoint-500/model.safetensors +3 -0
  28. checkpoint-500/optimizer.bin +3 -0
  29. checkpoint-500/pytorch_lora_weights.safetensors +3 -0
  30. checkpoint-500/random_states_0.pkl +3 -0
  31. checkpoint-500/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/1731878151.9085805/events.out.tfevents.1731878151.152bad1b73fe.202.1 +3 -0
  37. logs/text2image-fine-tune/1731878151.911212/hparams.yml +50 -0
  38. logs/text2image-fine-tune/1731878309.030499/events.out.tfevents.1731878309.152bad1b73fe.454.1 +3 -0
  39. logs/text2image-fine-tune/1731878309.0331604/hparams.yml +50 -0
  40. logs/text2image-fine-tune/events.out.tfevents.1731878151.152bad1b73fe.202.0 +3 -0
  41. logs/text2image-fine-tune/events.out.tfevents.1731878309.152bad1b73fe.454.0 +3 -0
  42. pytorch_lora_weights.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: runwayml/stable-diffusion-v1-5
3
+ library_name: diffusers
4
+ license: creativeml-openrail-m
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-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 - ButterChicken98/plantVillage-stableDiffusion
20
+ These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the ButterChicken98/platvillage_with_captions2 dataset. You can find some example images in the following.
21
+
22
+ ![img_0](./image_0.png)
23
+ ![img_1](./image_1.png)
24
+ ![img_2](./image_2.png)
25
+ ![img_3](./image_3.png)
26
+
27
+
28
+
29
+ ## Intended uses & limitations
30
+
31
+ #### How to use
32
+
33
+ ```python
34
+ # TODO: add an example code snippet for running this diffusion pipeline
35
+ ```
36
+
37
+ #### Limitations and bias
38
+
39
+ [TODO: provide examples of latent issues and potential remediations]
40
+
41
+ ## Training details
42
+
43
+ [TODO: describe the data used to train the model]
checkpoint-1000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:980136904dd1b1df284c49995307c1490ddf9062cd24db3daa13b56bc7bb3228
3
+ size 3441397096
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5978eb49144f10dec82fe54f89d2936a04da06cc843b595d034a7b5940785e9d
3
+ size 6592122
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81f799fd33f962d9280ec4384c34d1f86ba0a14ce23bd4479b9607a49393e7ab
3
+ size 3226184
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c8e534bfd54196bf267cd4a9686a65c91122b182e20131e8a859df6a7e370cc
3
+ size 14668
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:545821dc5493a874470abe7bed7760d7d04473241c9b6fb7974da5de059be251
3
+ size 1000
checkpoint-1500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1754e76add940aa046f6d4e60b8bf44096826f3066ccf89c66f58c064a8098d4
3
+ size 3441397096
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0239191e9cbba413b68c84596808033528584a773cb15ed38772d5f012262ce8
3
+ size 6592122
checkpoint-1500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d677dd61abd93b4e40b04767e311e0f9d517f155f3e30bda1e4a05e3946bea1
3
+ size 3226184
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84d4b698e50b238b7a185a6b572afe31216d71f0bc885f89072d641a3ec9520e
3
+ size 14668
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d21859f6d8ed1490d9b681089219a4e4924da8b3cbfeeb88803364b6a37a3253
3
+ size 1000
checkpoint-2000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cb2da3410aa45483cfe2233bdc70d5f0403b84f7a00b490cefc416d4f68c30e
3
+ size 3441397096
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9aff1abdea860c1886dbea9ee41c2920337079a35321acd7f93c4f4067f73e0d
3
+ size 6592122
checkpoint-2000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4b30b0e673b18a166aa7d0fb25685d145c238a11f7937458740ea9aa343e1f2
3
+ size 3226184
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42cc912d6b2fd7896cfd2a586b46e0ab3354923b61d9af370c3bba5ad4fa7a73
3
+ size 14668
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a139f59b5b69af0cda7ebfd38ce170a908e8c9c4a46f24561b5c7c8e15856f7a
3
+ size 1000
checkpoint-2500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51b39b88c96b9242f87ec23c5c699ce4f9cc1bba9f0c64c5944c7ea18421f5db
3
+ size 3441397096
checkpoint-2500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:601aa68042515366afcde908b7cedefb30f95af2fa0443a1e11bfa904978e53d
3
+ size 6592122
checkpoint-2500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:330b81c03dc076e0eb5135bd49e26819d45c042894ac76bcc6ee1e6158e2c636
3
+ size 3226184
checkpoint-2500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b2d68fe92b353407c2b921c8fdb8a0fdd926c56cfdb2cc66bb4f6527caa1250
3
+ size 14668
checkpoint-2500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74117f52c3ee01348dcb49b5de4ac469e83d2cca5127f65ca7ea1c3202bc596a
3
+ size 1000
checkpoint-3000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24c3bb1bf5e4fc1eb4996195cf358e1e55c31856d96585804276b957e943214d
3
+ size 3441397096
checkpoint-3000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c41faf8061eed3b0e3b009b8e8cede18506bf7fba5327d23bf7d2a70b78ef054
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:cf3d9175c1eaae571c5692735d7dd7e204eb5d666929ed3e49cb21400505da64
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:c7dfd325843d400b2211471a7de4e5a32533d5e5ac0f626459ccca3b0b556d60
3
+ size 14668
checkpoint-3000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa9aaa747f8bfe7bbd402fd01ce71c8b4e9bef64d71e2dae6c447447226609ab
3
+ size 1000
checkpoint-500/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10fb371d7ad2855266f629bee49537d8ca35e702a7830cbe580d441707bcfc6a
3
+ size 3441397096
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c057d77764a41db91338478408bd5be61655ba592f1825e987b1207bfef119da
3
+ size 6592122
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:044f1f181b9ea0478f6dcbf80aedad9f6b113e1ae3900fc0f960c9210ce53142
3
+ size 3226184
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e78c4f0877557e0ac738da7eaeb2f631599ab1da2362ce22bd7cf9ed68e1121
3
+ size 14668
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3670305799c0669149a15468e1d6085dd1012d5d39fa3c0e6b74165482d4115
3
+ size 1000
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/text2image-fine-tune/1731878151.9085805/events.out.tfevents.1731878151.152bad1b73fe.202.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6474d0f0324d748729c8e964ad70f82734fcf366aa044d1b23214573a830aee3
3
+ size 2237
logs/text2image-fine-tune/1731878151.911212/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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: ButterChicken98/platvillage_with_captions2
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: 6
31
+ num_validation_images: 4
32
+ output_dir: /content/plantVillage
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: 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: Totoro
50
+ variant: null
logs/text2image-fine-tune/1731878309.030499/events.out.tfevents.1731878309.152bad1b73fe.454.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43740ba992ffedca895056244848d64c4a2b60864e89ee010ce3e443e737b85d
3
+ size 2303
logs/text2image-fine-tune/1731878309.0331604/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: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 8
12
+ dataset_config_name: null
13
+ dataset_name: ButterChicken98/platvillage_with_captions2
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: false
17
+ hub_model_id: plantVillage-stableDiffusion
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: 3000
28
+ mixed_precision: null
29
+ noise_offset: 0
30
+ num_train_epochs: 2
31
+ num_validation_images: 4
32
+ output_dir: /content/plantVillage
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: null
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: Totoro
50
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1731878151.152bad1b73fe.202.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6b1a5fd813f548323bb95a8cdcf2118eb85a39869d32d304672a555d43e482f
3
+ size 328
logs/text2image-fine-tune/events.out.tfevents.1731878309.152bad1b73fe.454.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7153e54bede7af2f926f4111896977649b4dbd396ff57263a9bc42a36eaf9cb
3
+ size 4652150
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf3d9175c1eaae571c5692735d7dd7e204eb5d666929ed3e49cb21400505da64
3
+ size 3226184