Squaad AI commited on
Commit
437df82
1 Parent(s): 9babb46

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
4
+ instance_prompt: TOK
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - woman
14
+
15
+ These are LoRA adaption weights for [stabilityai/stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0). The weights were trained on the instance prompt "TOK" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
+
17
+
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:741dfee81a2f2a501577642a82ee5085f6cda93d67ef0318fabdcb7c92db734d
3
+ size 225713466
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5faa66da0fbac2cc865b31d9db6d57901f33c5da98b20390fd435c5175278da9
3
+ size 112769968
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8dd1820099fdf972b4880c99f24265a43a681a6f8842475c55243d3eeb04b38c
3
+ size 14344
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d8fdcd0311eba9854fff738038ed4c1a269832665b4d88ba4e4e3d02a1a7e0e
3
+ size 988
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1f26516688b8724db1aac8f799259f80da0817b95966b5e91086005c8918d51
3
+ size 1000
logs/dreambooth-lora-sd-xl/1711765713.624255/events.out.tfevents.1711765713.r-amazonaws-sp-b-lora-k7x5xb5a-ea13c-veqc6.3395.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ef6cba8250127deee70fb1ba846317e3c11322d2f3bd769ae91047e60838022
3
+ size 3146
logs/dreambooth-lora-sd-xl/1711765713.626018/hparams.yml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: null
9
+ center_crop: false
10
+ checkpointing_steps: 1000
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: null
17
+ do_edm_style_training: false
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: /home/user/app/training_data/woman
25
+ instance_prompt: TOK
26
+ learning_rate: 0.0004
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 1000
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 50
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: /home/user/app/experiments/woman
41
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
42
+ pretrained_vae_model_name_or_path: null
43
+ prior_generation_precision: null
44
+ prior_loss_weight: 1.0
45
+ prodigy_beta3: null
46
+ prodigy_decouple: true
47
+ prodigy_safeguard_warmup: true
48
+ prodigy_use_bias_correction: true
49
+ push_to_hub: false
50
+ random_flip: false
51
+ rank: 64
52
+ repeats: 1
53
+ report_to: tensorboard
54
+ resolution: 1024
55
+ resume_from_checkpoint: null
56
+ revision: null
57
+ sample_batch_size: 4
58
+ scale_lr: false
59
+ seed: 11230
60
+ snr_gamma: null
61
+ text_encoder_lr: 5.0e-06
62
+ train_batch_size: 1
63
+ train_text_encoder: false
64
+ use_8bit_adam: false
65
+ use_dora: false
66
+ validation_epochs: 250
67
+ validation_prompt: A photo of TOK
68
+ variant: null
69
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1711765713.r-amazonaws-sp-b-lora-k7x5xb5a-ea13c-veqc6.3395.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e2d3687b53f8f0b8a8b37d34023c7dc42ad10f1248532a17abcc1ef13ace03f
3
+ size 108514
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5faa66da0fbac2cc865b31d9db6d57901f33c5da98b20390fd435c5175278da9
3
+ size 112769968
train.sh ADDED
@@ -0,0 +1 @@
 
 
1
+ accelerate launch train_dreambooth_b-lora_sdxl.py --pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0 --instance_data_dir=/home/user/app/training_data/woman --output_dir=/home/user/app/experiments/woman --instance_prompt="TOK" --resolution=1024 --rank=64 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0004 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=1000 --validation_prompt="A photo of TOK" --validation_epochs=250 --seed=11230 --mixed_precision fp16