Squaad AI commited on
Commit
c6cc0b8
1 Parent(s): 893b1f9

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 TOK character
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ inference: true
12
+ ---
13
+ # LoRA DreamBooth - kirito
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 TOK character" 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:102d89dee9d56f365a2cc232fd90da45e7851aa56a61773245a6fb5d6d9c6356
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:2c8fe5885612bd852e8a2ad9cd869d2122c78a21110de0c9758ff77d7cbea853
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:1e52c286b775409ff16d41fa374b264e3bb980089203e6017fc93e684e50928d
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:6a06a1d190ebe3d2f59b87075958832ab264ff51bb09474ff935bfa56dcf4374
3
+ size 1000
logs/dreambooth-lora-sd-xl/1711812733.5171144/events.out.tfevents.1711812733.r-amazonaws-sp-b-lora-y8vwmifr-ea63e-erjso.838.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f138cf3063dfdac912af0d781ca1e64cfaae2fef74167ca09fb8d7dbe2761bb
3
+ size 3161
logs/dreambooth-lora-sd-xl/1711812733.51888/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/kirito
25
+ instance_prompt: A TOK character
26
+ learning_rate: 5.0e-05
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: 77
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: /home/user/app/experiments/kirito
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: 8746
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 TOK character
68
+ variant: null
69
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1711812733.r-amazonaws-sp-b-lora-y8vwmifr-ea63e-erjso.838.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12566eae35f66fac052515746f72fce285c34c60b22d3cbd9295551e128ce6bd
3
+ size 108514
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c8fe5885612bd852e8a2ad9cd869d2122c78a21110de0c9758ff77d7cbea853
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/kirito --output_dir=/home/user/app/experiments/kirito --instance_prompt="A TOK character" --resolution=1024 --rank=64 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=5e-05 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=1000 --validation_prompt="A TOK character" --validation_epochs=250 --seed=8746 --mixed_precision fp16