Hugging Face's logo

lora-library
/
simbatheog

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files Metrics Community
ysharma HF staff commited on
Commit
f1f86c4
•
1 Parent(s): 80cf3e9

Upload . with huggingface_hub

Browse files
Files changed (40) hide show
  1. README.md +22 -0
  2. checkpoint-1000/custom_checkpoint_0.pkl +3 -0
  3. checkpoint-1000/optimizer.bin +3 -0
  4. checkpoint-1000/pytorch_model.bin +3 -0
  5. checkpoint-1000/random_states_0.pkl +3 -0
  6. checkpoint-1000/scaler.pt +3 -0
  7. checkpoint-1000/scheduler.bin +3 -0
  8. checkpoint-1500/custom_checkpoint_0.pkl +3 -0
  9. checkpoint-1500/optimizer.bin +3 -0
  10. checkpoint-1500/pytorch_model.bin +3 -0
  11. checkpoint-1500/random_states_0.pkl +3 -0
  12. checkpoint-1500/scaler.pt +3 -0
  13. checkpoint-1500/scheduler.bin +3 -0
  14. checkpoint-2000/custom_checkpoint_0.pkl +3 -0
  15. checkpoint-2000/optimizer.bin +3 -0
  16. checkpoint-2000/pytorch_model.bin +3 -0
  17. checkpoint-2000/random_states_0.pkl +3 -0
  18. checkpoint-2000/scaler.pt +3 -0
  19. checkpoint-2000/scheduler.bin +3 -0
  20. checkpoint-2500/custom_checkpoint_0.pkl +3 -0
  21. checkpoint-2500/optimizer.bin +3 -0
  22. checkpoint-2500/pytorch_model.bin +3 -0
  23. checkpoint-2500/random_states_0.pkl +3 -0
  24. checkpoint-2500/scaler.pt +3 -0
  25. checkpoint-2500/scheduler.bin +3 -0
  26. checkpoint-500/custom_checkpoint_0.pkl +3 -0
  27. checkpoint-500/optimizer.bin +3 -0
  28. checkpoint-500/pytorch_model.bin +3 -0
  29. checkpoint-500/random_states_0.pkl +3 -0
  30. checkpoint-500/scaler.pt +3 -0
  31. checkpoint-500/scheduler.bin +3 -0
  32. logs/dreambooth-lora/1675590864.9572244/events.out.tfevents.1675590864.s-ysharma-lora-dreambooth-training-ui-90852-6fc688cf68-mnphc.98.1 +3 -0
  33. logs/dreambooth-lora/1675590864.9588535/hparams.yml +48 -0
  34. logs/dreambooth-lora/events.out.tfevents.1675590864.s-ysharma-lora-dreambooth-training-ui-90852-6fc688cf68-mnphc.98.0 +3 -0
  35. pytorch_lora_weights.bin +3 -0
  36. test_images/image_0.png +0 -0
  37. test_images/image_1.png +0 -0
  38. test_images/image_2.png +0 -0
  39. test_images/image_3.png +0 -0
  40. train.sh +1 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ base_model: stabilityai/stable-diffusion-2-1-base
4
+ instance_prompt: simbatheog
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - simbatheog
14
+
15
+ These are LoRA adaption weights for [stabilityai/stable-diffusion-2-1-base](https://huggingface.co/stabilityai/stable-diffusion-2-1-base). The weights were trained on the instance prompt "simbatheog" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
+
17
+ Test prompt: disney lion cub
18
+ ![image_0](test_images/image_0.png)
19
+ ![image_1](test_images/image_1.png)
20
+ ![image_2](test_images/image_2.png)
21
+ ![image_3](test_images/image_3.png)
22
+
checkpoint-1000/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72ec4bb4276296634fdfc1d4c835d3041e9c023a872b67f856d44f1b0ba8b145
3
+ size 3418585
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a908abe6acbe7608e7ea391f292638727438dcada4a63870e5dfe11a987bf41
3
+ size 3493125
checkpoint-1000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a08aaab71707fe2acde6b4f16c3490d571a413f9c295b8735175bc88cd2a115d
3
+ size 3417037
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7102b4014425864633a6405beea3b165fa9ba57acbce9aeb9ee604817da984d9
3
+ size 14599
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:963c3085720a3a3ca1d178559a0f2dc5d5b36fa395a4663f0194bfac4a754038
3
+ size 563
checkpoint-1500/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:155d7c9707d9c291c3383223511a23d906aad4515fc870097c92ea58cffd2204
3
+ size 3418585
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2547a44db707c6b2d6c325c0665785078a6c906f296af7c2b260897642a10321
3
+ size 3493125
checkpoint-1500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce0a784cdd198f00820f7d776d1e0ae9467a955f039047a016838f3d40ffba8f
3
+ size 3417037
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:173052faff7b69b882d519ebd399b22acd6062cef85982a2579b3452d1171264
3
+ size 14599
checkpoint-1500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9c8fd3d61799f474b16755bb2883096b88d67ad1ceea7570f8a2ef5f71c197e
3
+ size 563
checkpoint-2000/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:155d7c9707d9c291c3383223511a23d906aad4515fc870097c92ea58cffd2204
3
+ size 3418585
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2547a44db707c6b2d6c325c0665785078a6c906f296af7c2b260897642a10321
3
+ size 3493125
checkpoint-2000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce0a784cdd198f00820f7d776d1e0ae9467a955f039047a016838f3d40ffba8f
3
+ size 3417037
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:265ecdcd24195a4cc080fb52c8e5822ebc2d34fc50269dd2fad7183c3eec82d5
3
+ size 14599
checkpoint-2000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:456e0f82e1f69c1c7c3f294d8d9ee5a27f11a3af5067518517fb2354b1dae333
3
+ size 563
checkpoint-2500/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:155d7c9707d9c291c3383223511a23d906aad4515fc870097c92ea58cffd2204
3
+ size 3418585
checkpoint-2500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2547a44db707c6b2d6c325c0665785078a6c906f296af7c2b260897642a10321
3
+ size 3493125
checkpoint-2500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce0a784cdd198f00820f7d776d1e0ae9467a955f039047a016838f3d40ffba8f
3
+ size 3417037
checkpoint-2500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f702039f1a4ff051d08df52b8dc47116a1c90fdf4ce863341072fb60c5580f3
3
+ size 14599
checkpoint-2500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-2500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e0dabfefc73f3aff42da866cecd1c0190eae004d88abd0a72ba68c52491f31f
3
+ size 563
checkpoint-500/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c93551fc57c84bdcf75c60b42f3d4bf7f21632831eab1eab3ea9e57435c33d5
3
+ size 3418585
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5b0f117ecbccb1c668f39f2d4aa565128bc2bcf9ed338537f36da2b1120bb7b
3
+ size 3493125
checkpoint-500/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fbf57db79f66d854904c3c0338d6b7196f1416e7fc2d5b9deee70d5f6eb873b
3
+ size 3417037
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d84f1bb4dfdf0589fe54beab793b2604a54ec2c97850729233741e8032a02f4c
3
+ size 14599
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:62783ff6f0ae2c2ccc177b097e430c03940c5e70c38a191df273c5cf7ab1227c
3
+ size 563
logs/dreambooth-lora/1675590864.9572244/events.out.tfevents.1675590864.s-ysharma-lora-dreambooth-training-ui-90852-6fc688cf68-mnphc.98.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2d9e3144ef9c72bcedd014f2310892d0586547c009f103f2a38795d1409e4bd
3
+ size 2440
logs/dreambooth-lora/1675590864.9588535/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
7
+ checkpointing_steps: 500
8
+ class_data_dir: null
9
+ class_prompt: null
10
+ dataloader_num_workers: 0
11
+ enable_xformers_memory_efficient_attention: false
12
+ gradient_accumulation_steps: 1
13
+ gradient_checkpointing: false
14
+ hub_model_id: null
15
+ hub_token: null
16
+ instance_data_dir: /home/user/app/training_data/simbatheog
17
+ instance_prompt: simbatheog
18
+ learning_rate: 0.0001
19
+ local_rank: -1
20
+ logging_dir: logs
21
+ lr_num_cycles: 1
22
+ lr_power: 1.0
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ max_grad_norm: 1.0
26
+ max_train_steps: 2500
27
+ mixed_precision: fp16
28
+ num_class_images: 100
29
+ num_train_epochs: 625
30
+ num_validation_images: 4
31
+ output_dir: /home/user/app/experiments/simbatheog
32
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1-base
33
+ prior_generation_precision: null
34
+ prior_loss_weight: 1.0
35
+ push_to_hub: false
36
+ report_to: tensorboard
37
+ resolution: 512
38
+ resume_from_checkpoint: null
39
+ revision: null
40
+ sample_batch_size: 4
41
+ scale_lr: false
42
+ seed: 0
43
+ tokenizer_name: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: true
46
+ validation_epochs: 100
47
+ validation_prompt: disney lion cub
48
+ with_prior_preservation: false
logs/dreambooth-lora/events.out.tfevents.1675590864.s-ysharma-lora-dreambooth-training-ui-90852-6fc688cf68-mnphc.98.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cf76804739ad79591cf0f14713979a0f9294618d2fe720895308ed0ea4dc866
3
+ size 5480816
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4357a54f28600cba0da6aa842c11b7429761d87fef20380998b9f2dda0f26e4
3
+ size 3418843
test_images/image_0.png ADDED
test_images/image_1.png ADDED
test_images/image_2.png ADDED
test_images/image_3.png ADDED
train.sh ADDED
@@ -0,0 +1 @@
 
 
1
+ accelerate launch train_dreambooth_lora.py --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base --instance_data_dir=/home/user/app/training_data/simbatheog --output_dir=/home/user/app/experiments/simbatheog --instance_prompt="simbatheog" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=2500 --checkpointing_steps=500 --validation_prompt="disney lion cub" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam