Hugging Face's logo

taraxis
/
nekotest1-1

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files Metrics Community
taraxis commited on
Commit
1c8c724
•
1 Parent(s): 0570bb2

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-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-400/custom_checkpoint_0.pkl +3 -0
  15. checkpoint-400/optimizer.bin +3 -0
  16. checkpoint-400/pytorch_model.bin +3 -0
  17. checkpoint-400/random_states_0.pkl +3 -0
  18. checkpoint-400/scaler.pt +3 -0
  19. checkpoint-400/scheduler.bin +3 -0
  20. checkpoint-600/custom_checkpoint_0.pkl +3 -0
  21. checkpoint-600/optimizer.bin +3 -0
  22. checkpoint-600/pytorch_model.bin +3 -0
  23. checkpoint-600/random_states_0.pkl +3 -0
  24. checkpoint-600/scaler.pt +3 -0
  25. checkpoint-600/scheduler.bin +3 -0
  26. checkpoint-800/custom_checkpoint_0.pkl +3 -0
  27. checkpoint-800/optimizer.bin +3 -0
  28. checkpoint-800/pytorch_model.bin +3 -0
  29. checkpoint-800/random_states_0.pkl +3 -0
  30. checkpoint-800/scaler.pt +3 -0
  31. checkpoint-800/scheduler.bin +3 -0
  32. logs/dreambooth-lora/1675997662.2605665/events.out.tfevents.1675997662.s-taraxis-lora-dreambooth-training-ui-abc3c-59d97f84b6-j7ppf.155.1 +3 -0
  33. logs/dreambooth-lora/1675997662.262569/hparams.yml +48 -0
  34. logs/dreambooth-lora/events.out.tfevents.1675997662.s-taraxis-lora-dreambooth-training-ui-abc3c-59d97f84b6-j7ppf.155.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: nekotst
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - nekotest1-1
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 "nekotst" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
+
17
+ Test prompt: a tabby cat
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:d4ea048819ad74c8c901199a6e68c4314e20ec87f206d73ab1c74bffa88c9c80
3
+ size 3418585
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad004933592c7e26bb66669cf40a0b32d68bb3aabaa4b4c5b580d87a610ee2e4
3
+ size 3492869
checkpoint-1000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5aa352eb51588557b8afbcc97766b6fe05aa0a50cbae366c4279e59d7371805
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:c9c20ea63922f8fbe97a5443f0a8cab608c7784bc0892a3ff5f8ba5887078df7
3
+ size 14663
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd24ac9bb08b8acc6b749bfb3328596e929bc788c6fb96fcf34442d6d9831ea9
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:d4ea048819ad74c8c901199a6e68c4314e20ec87f206d73ab1c74bffa88c9c80
3
+ size 3418585
checkpoint-200/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad004933592c7e26bb66669cf40a0b32d68bb3aabaa4b4c5b580d87a610ee2e4
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:e5aa352eb51588557b8afbcc97766b6fe05aa0a50cbae366c4279e59d7371805
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:7ff556c9087a7d3358adc1f844c657606858901f312c6e1e62a09bd53526d695
3
+ size 14663
checkpoint-200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68189287a9e8493bed245fa00d8cabf50d0feb8ed9f5419568c185e9b8dfb87c
3
+ size 557
checkpoint-200/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e089907bdb3d8eb575c35589953c5564e01d7aa13d43fc67730d3f7a4a3499bb
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:d4ea048819ad74c8c901199a6e68c4314e20ec87f206d73ab1c74bffa88c9c80
3
+ size 3418585
checkpoint-400/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad004933592c7e26bb66669cf40a0b32d68bb3aabaa4b4c5b580d87a610ee2e4
3
+ size 3492869
checkpoint-400/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5aa352eb51588557b8afbcc97766b6fe05aa0a50cbae366c4279e59d7371805
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:3f36976507255dc72dac1db36b14310f6b53a35492ad4e45d0e3899429ee4885
3
+ size 14663
checkpoint-400/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-400/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1816c3518fba5280b8d446f53a7a78cb091fb0abfbf56998eb7ab2ec499596cc
3
+ size 563
checkpoint-600/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4ea048819ad74c8c901199a6e68c4314e20ec87f206d73ab1c74bffa88c9c80
3
+ size 3418585
checkpoint-600/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad004933592c7e26bb66669cf40a0b32d68bb3aabaa4b4c5b580d87a610ee2e4
3
+ size 3492869
checkpoint-600/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5aa352eb51588557b8afbcc97766b6fe05aa0a50cbae366c4279e59d7371805
3
+ size 3417037
checkpoint-600/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43bd26f82eb26f8ed44510144f0471c2141bccd83a21542f586255600a41888a
3
+ size 14663
checkpoint-600/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-600/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2665a7c384abc70b2477397d0702e8a28551d53eea8baf81d5952af4d95f3974
3
+ size 563
checkpoint-800/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4ea048819ad74c8c901199a6e68c4314e20ec87f206d73ab1c74bffa88c9c80
3
+ size 3418585
checkpoint-800/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad004933592c7e26bb66669cf40a0b32d68bb3aabaa4b4c5b580d87a610ee2e4
3
+ size 3492869
checkpoint-800/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5aa352eb51588557b8afbcc97766b6fe05aa0a50cbae366c4279e59d7371805
3
+ size 3417037
checkpoint-800/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ffa79cbadac79e34720c8fc1605cfa8b43cad503406dcecf56cf1efbaa70d83
3
+ size 14663
checkpoint-800/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
3
+ size 557
checkpoint-800/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a8b0e01f93edaffa41e24a7bf9c6f2eda73bf2b29cd8fd2d90979b2023dbb71
3
+ size 563
logs/dreambooth-lora/1675997662.2605665/events.out.tfevents.1675997662.s-taraxis-lora-dreambooth-training-ui-abc3c-59d97f84b6-j7ppf.155.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:520bdbabaf2cd49436346609d09430a18029fb15ea89db79eb4c44bf7ffb08d4
3
+ size 2435
logs/dreambooth-lora/1675997662.262569/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: 200
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/nekotest1-1
17
+ instance_prompt: nekotst
18
+ learning_rate: 0.0005
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: 1000
27
+ mixed_precision: fp16
28
+ num_class_images: 100
29
+ num_train_epochs: 36
30
+ num_validation_images: 4
31
+ output_dir: /home/user/app/experiments/nekotest1-1
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: 80085
43
+ tokenizer_name: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: true
46
+ validation_epochs: 100
47
+ validation_prompt: a tabby cat
48
+ with_prior_preservation: false
logs/dreambooth-lora/events.out.tfevents.1675997662.s-taraxis-lora-dreambooth-training-ui-abc3c-59d97f84b6-j7ppf.155.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b0b46450e90c76fd4f8ed3c073404b84fdbbf009340d2d814958dc756d86c98
3
+ size 3064195
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9cb4a0d903d960657855ec3283877bdfd808a375bb12d08d247b828257f1215
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/nekotest1-1 --output_dir=/home/user/app/experiments/nekotest1-1 --instance_prompt="nekotst" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0005 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=200 --validation_prompt="a tabby cat" --validation_epochs=100 --seed=80085 --mixed_precision fp16 --use_8bit_adam