Hugging Face's logo

lora-library
/
underwater-fish

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files Metrics Community
vaseto commited on
Commit
4aa96f7
•
1 Parent(s): 110dc89

Upload . with huggingface_hub

Browse files
Files changed (40) hide show
  1. README.md +22 -0
  2. checkpoint-100/custom_checkpoint_0.pkl +3 -0
  3. checkpoint-100/optimizer.bin +3 -0
  4. checkpoint-100/pytorch_model.bin +3 -0
  5. checkpoint-100/random_states_0.pkl +3 -0
  6. checkpoint-100/scaler.pt +3 -0
  7. checkpoint-100/scheduler.bin +3 -0
  8. checkpoint-200/custom_checkpoint_0.pkl +3 -0
  9. checkpoint-200/optimizer.bin +3 -0
  10. checkpoint-200/pytorch_model.bin +3 -0
  11. checkpoint-200/random_states_0.pkl +3 -0
  12. checkpoint-200/scaler.pt +3 -0
  13. checkpoint-200/scheduler.bin +3 -0
  14. checkpoint-300/custom_checkpoint_0.pkl +3 -0
  15. checkpoint-300/optimizer.bin +3 -0
  16. checkpoint-300/pytorch_model.bin +3 -0
  17. checkpoint-300/random_states_0.pkl +3 -0
  18. checkpoint-300/scaler.pt +3 -0
  19. checkpoint-300/scheduler.bin +3 -0
  20. checkpoint-400/custom_checkpoint_0.pkl +3 -0
  21. checkpoint-400/optimizer.bin +3 -0
  22. checkpoint-400/pytorch_model.bin +3 -0
  23. checkpoint-400/random_states_0.pkl +3 -0
  24. checkpoint-400/scaler.pt +3 -0
  25. checkpoint-400/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/1700843966.7397125/events.out.tfevents.1700843966.r-vaseto-lora-dreambooth-training-ui-test-msc5kfke-13886-xjmxd.103.1 +3 -0
  33. logs/dreambooth-lora/1700843966.74149/hparams.yml +48 -0
  34. logs/dreambooth-lora/events.out.tfevents.1700843966.r-vaseto-lora-dreambooth-training-ui-test-msc5kfke-13886-xjmxd.103.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: mp_nderwater
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - underwater-fish
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 "mp_nderwater" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
+
17
+ Test prompt: 3d underwater fish
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-100/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5d51e77083e97faea32a7e7723bb2a3f46939ad5502b4d627cf20bb51830181
3
+ size 3418585
checkpoint-100/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12789fd6374163ce5e81f99f5cb4b79f8a053ef15051b6d80d4a7b8483b1daa7
3
+ size 3492869
checkpoint-100/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21088238ae35a5cf984c0121f832aefe94949940f4c30b0aec49b48c45a39cfd
3
+ size 3417037
checkpoint-100/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3c89a8ce1131e762c8587de119f2d24d20459b83d0d6e78c5c16ba4b93acbb5
3
+ size 14599
checkpoint-100/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfa44e8523f62833816d29aa6c576eaa7783e3bbdb3e132e248b1d8aaee6132b
3
+ size 557
checkpoint-100/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:906bdd391189fd16d3160fd1e198eb53687dae580f930cf3f64365609fd5f606
3
+ size 563
checkpoint-200/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:537843ebb7146fa3b1d8124bb2673bc55440f39e4887624d2f9308370d3adf06
3
+ size 3418585
checkpoint-200/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6e1117a47316f72fb87fd2615c4c7d1407dec424bdb337bd1890fd5097d8dbe
3
+ size 3492869
checkpoint-200/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcbb0830487ed4ebec344b5b40c054c9fafe921641784f6f1ba66294d2bf5615
3
+ size 3417037
checkpoint-200/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0eeeeecf22d6e590b928b8b50d6f199196bb3c03c37e1ff0150f4a08702b65fa
3
+ size 14599
checkpoint-200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fccf0f9be1bb8f24861e4393745b3e09cc2687125a69e3757955fb0f0925ea5
3
+ size 557
checkpoint-200/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2565f1f317aadcb95557955732a7a05efb023cacf90620d89fcb0c7ac0f8e04
3
+ size 563
checkpoint-300/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a8e4f763dbcac25bdda19cc92b5ab6627793d4dc3ee11dba570dba06f785176
3
+ size 3418585
checkpoint-300/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b679d65eb13bbeb2bde086d9f4c789016a2782f810faa0df5478ae4633f61771
3
+ size 3493125
checkpoint-300/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:331c104044d339eff729d86bf55994978c6f457503287ccd5985492bdae22a0e
3
+ size 3417037
checkpoint-300/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e54fcb6e49ce98eb34521180b9b54b58ef7261bb97eee60a5368889de1c0c43
3
+ size 14599
checkpoint-300/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efafd90182e3d39d1b7c4a686f86e5913f5abc094dc3e2f827a6d479c6cef247
3
+ size 557
checkpoint-300/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a84fb1fc02dc9adb8c44beeab2c46fbc41e10a07e99dad3fcd85350ca2bda9d5
3
+ size 563
checkpoint-400/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59600ac98a9d7d141e92c43622c30a93ec5ada53695f11328802adcfa57a73d7
3
+ size 3418585
checkpoint-400/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7262f019a5dc97ae9f33bf67c094b01a9afdc6feba6007feb6adec448afd22af
3
+ size 3493125
checkpoint-400/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e415a310799a6ae750872e4cccef16cb8dd2e5e50cb9ced093eef88db61c1a7d
3
+ size 3417037
checkpoint-400/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85b7b1bd6b5f84ac6b690f725318d78b5090469072c4ec96350f241bf059bf8b
3
+ size 14599
checkpoint-400/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc40a4be6a52cee4d7658df4041e660ffa02b0d8b5bd143bb8bb397f7b71b1a5
3
+ size 557
checkpoint-400/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e33b45756ec30d1126f6571b94c7435f6d9cb7b6d5e5e650c81523775e7216f
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:48607a36c0a566ead4bb0811e9535a129af0d2c5aead4f1f6f5aa6f9b6102b6f
3
+ size 3418585
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:331fcc35edf629b86ce692ab63606281a124de53efaaffa4a09c13d513e588a5
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:b24765efc79b354d9bca5922b2330cdfd9df3abc64e59f31b1403f467fde1380
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:fb6b69404ebb390032a82927de31a8b578038f5b1987de4e2231bbd38ccfe340
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/1700843966.7397125/events.out.tfevents.1700843966.r-vaseto-lora-dreambooth-training-ui-test-msc5kfke-13886-xjmxd.103.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1298bb575490cc0bd906d6e8483080b6c5c0844f455473512bfa94b768b9d424
3
+ size 2455
logs/dreambooth-lora/1700843966.74149/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: 100
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/underwater-fish
17
+ instance_prompt: mp_nderwater
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: 500
27
+ mixed_precision: fp16
28
+ num_class_images: 100
29
+ num_train_epochs: 84
30
+ num_validation_images: 4
31
+ output_dir: /home/user/app/experiments/underwater-fish
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: 768
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: 3d underwater fish
48
+ with_prior_preservation: false
logs/dreambooth-lora/events.out.tfevents.1700843966.r-vaseto-lora-dreambooth-training-ui-test-msc5kfke-13886-xjmxd.103.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08497282222c9f2059f1b68ee1d386dc63d5f0a18672d447fd9d233e14f104ea
3
+ size 3666266
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:693969c8a7808d5b469aad7cd62683dce1edbbc7439c9b232549f6b3c13eea58
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/underwater-fish --output_dir=/home/user/app/experiments/underwater-fish --instance_prompt="mp_nderwater" --resolution=768 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=500 --checkpointing_steps=100 --validation_prompt="3d underwater fish" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam