Squaad AI commited on
Commit
1155f2c
·
verified ·
1 Parent(s): 0d84eda

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: A photo of 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 - kirito2
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 "A photo of 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:ff6abeb65779e2c8ca2bb32c209d184a13e59ecbe2623984cc62554ddeff2598
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:a22680cece0694e1ea48dda114cff7a0c6f7f8474a07dce78c6b26bf03a7401b
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:3535289416387ca7167bc6c777d429a03e2100570011080e9f37c39540a68a97
3
+ size 14280
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/1711814198.996586/events.out.tfevents.1711814198.r-amazonaws-sp-b-lora-y8vwmifr-ea63e-erjso.1134.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d2a54e71fad2b953b0f7b1ffb785aea3b46d3d86492bd541d9821ce0f69ff71
3
+ size 3161
logs/dreambooth-lora-sd-xl/1711814198.9983742/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/kirito2
25
+ instance_prompt: A photo of 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/kirito2
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: 15525
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.1711814198.r-amazonaws-sp-b-lora-y8vwmifr-ea63e-erjso.1134.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30f53de53136e34b728dcc29a68b3d24fbea652adfbb4977ee420dc48acca9a7
3
+ size 108514
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a22680cece0694e1ea48dda114cff7a0c6f7f8474a07dce78c6b26bf03a7401b
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/kirito2 --output_dir=/home/user/app/experiments/kirito2 --instance_prompt="A photo of 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=15525 --mixed_precision fp16