Reversorie commited on
Commit
a357c83
1 Parent(s): 6aa795c

Upload . with huggingface_hub

Browse files
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: fabric design of small flowers
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - v1
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 "fabric design of small flowers" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
+
17
+ Test prompt: v1
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:b223b0aa4a7fae87097c966f33a0245a924b84bdbb0b31d384d7c854411cf50f
3
+ size 3418585
checkpoint-100/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3be2c5c5b71e4dcea964f0e584a154969aff4e6d6e29004d0e2e9ca2b32f7698
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:d9078b0d36df08a93c059ba97db275f92d94fdde24a01dc8ef9b3ac383e3a023
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:be32873b28d2d142aca149a6b92f9e3e8596e41ad0d3be5f261c9ad1ed888f55
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
logs/dreambooth-lora/1678578414.173294/events.out.tfevents.1678578414.s-reversorie-lora-dreambooth-training-ui-147db-59fc7c8567-dh5bk.87.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f550a2c71205ca7136f3357804fe968edec6ad2a911a0a93e28f2d67fd6f3529
3
+ size 2431
logs/dreambooth-lora/1678578414.175752/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/v1
17
+ instance_prompt: fabric design of small flowers
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: 100
27
+ mixed_precision: fp16
28
+ num_class_images: 100
29
+ num_train_epochs: 13
30
+ num_validation_images: 4
31
+ output_dir: /home/user/app/experiments/v1
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: v1
48
+ with_prior_preservation: false
logs/dreambooth-lora/events.out.tfevents.1678578414.s-reversorie-lora-dreambooth-training-ui-147db-59fc7c8567-dh5bk.87.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ff33ca715298f7fc40b1ba3ab494239d8107336216a9f6eb52159672808aa8a
3
+ size 4000621
pytorch_lora_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ea028f482b096107731258a6bc0de48e95a1fbab0330604b0598ec8922ed23d
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/v1 --output_dir=/home/user/app/experiments/v1 --instance_prompt="fabric design of small flowers" --resolution=768 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=100 --checkpointing_steps=100 --validation_prompt="v1" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam