BruBel commited on
Commit
90ec7ee
1 Parent(s): 38d8cc1

End of training

Browse files
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ image_0.png filter=lfs diff=lfs merge=lfs -text
37
+ image_1.png filter=lfs diff=lfs merge=lfs -text
38
+ image_2.png filter=lfs diff=lfs merge=lfs -text
39
+ image_3.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - stable-diffusion-xl
4
+ - stable-diffusion-xl-diffusers
5
+ - diffusers-training
6
+ - text-to-image
7
+ - diffusers
8
+ - lora
9
+ - template:sd-lora
10
+ widget:
11
+
12
+ - text: 'A Poro situated in a red sky, war scenario. The Poro wears a brown cape. Region of Shurima. In <s0><s1> style.'
13
+ output:
14
+ url:
15
+ "image_0.png"
16
+
17
+ - text: 'A Poro situated in a red sky, war scenario. The Poro wears a brown cape. Region of Shurima. In <s0><s1> style.'
18
+ output:
19
+ url:
20
+ "image_1.png"
21
+
22
+ - text: 'A Poro situated in a red sky, war scenario. The Poro wears a brown cape. Region of Shurima. In <s0><s1> style.'
23
+ output:
24
+ url:
25
+ "image_2.png"
26
+
27
+ - text: 'A Poro situated in a red sky, war scenario. The Poro wears a brown cape. Region of Shurima. In <s0><s1> style.'
28
+ output:
29
+ url:
30
+ "image_3.png"
31
+
32
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
33
+ instance_prompt: In <s0><s1> style.
34
+ license: openrail++
35
+ ---
36
+
37
+ # SDXL LoRA DreamBooth - BruBel/lor-style-training-v6
38
+
39
+ <Gallery />
40
+
41
+ ## Model description
42
+
43
+ ### These are BruBel/lor-style-training-v6 LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
44
+
45
+ ## Download model
46
+
47
+ ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
48
+
49
+ - **LoRA**: download **[`lor-style-training-v6.safetensors` here 💾](/BruBel/lor-style-training-v6/blob/main/lor-style-training-v6.safetensors)**.
50
+ - Place it on your `models/Lora` folder.
51
+ - On AUTOMATIC1111, load the LoRA by adding `<lora:lor-style-training-v6:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
52
+ - *Embeddings*: download **[`lor-style-training-v6_emb.safetensors` here 💾](/BruBel/lor-style-training-v6/blob/main/lor-style-training-v6_emb.safetensors)**.
53
+ - Place it on it on your `embeddings` folder
54
+ - Use it by adding `lor-style-training-v6_emb` to your prompt. For example, `In lor-style-training-v6_emb style.`
55
+ (you need both the LoRA and the embeddings as they were trained together for this LoRA)
56
+
57
+
58
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
59
+
60
+ ```py
61
+ from diffusers import AutoPipelineForText2Image
62
+ import torch
63
+ from huggingface_hub import hf_hub_download
64
+ from safetensors.torch import load_file
65
+
66
+ pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
67
+ pipeline.load_lora_weights('BruBel/lor-style-training-v6', weight_name='pytorch_lora_weights.safetensors')
68
+ embedding_path = hf_hub_download(repo_id='BruBel/lor-style-training-v6', filename='lor-style-training-v6_emb.safetensors', repo_type="model")
69
+ state_dict = load_file(embedding_path)
70
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
71
+ pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
72
+
73
+ image = pipeline('A Poro situated in a red sky, war scenario. The Poro wears a brown cape. Region of Shurima. In <s0><s1> style.').images[0]
74
+ ```
75
+
76
+ For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
77
+
78
+ ## Trigger words
79
+
80
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
81
+
82
+ to trigger concept `LEGENDSOFRUNETERRA` → use `<s0><s1>` in your prompt
83
+
84
+
85
+
86
+ ## Details
87
+ All [Files & versions](/BruBel/lor-style-training-v6/tree/main).
88
+
89
+ The weights were trained using [🧨 diffusers Advanced Dreambooth Training Script](https://github.com/huggingface/diffusers/blob/main/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py).
90
+
91
+ LoRA for the text encoder was enabled. False.
92
+
93
+ Pivotal tuning was enabled: True.
94
+
95
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
96
+
97
+
checkpoint-10000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b71b55b93964093a8c2b22e3a08aa06080d018fa390a9342325c4a1bfa06df13
3
+ size 1182072760
checkpoint-10000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c74650a6dde630e075b591a1626250444a09993ce8b1108c7ef889a6faabeb23
3
+ size 185964888
checkpoint-10000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:201045b36031a1cf06acac30fc15fee42b1ad3972f9824734e19a9c9b3f6ba25
3
+ size 14408
checkpoint-10000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcc150ecb5350ee944d6ece70445c822aed2b29c2785d6660ba82247100862dc
3
+ size 1064
checkpoint-20000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a6c91fc083200da6f011132d213ad60b6ed98000c69c1219eae0f7faf730fc3
3
+ size 1182072760
checkpoint-20000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f94d7514676a8bdb2481af960f63e62ba5c8b954572ed46ae925f3be59e2fa10
3
+ size 185964888
checkpoint-20000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ed548937b7052ff0b9fcf9ce14156e943aebcf9c8d33a5d649b5807d1152837
3
+ size 14408
checkpoint-20000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f615bccb0e7a5fa7fda853938ae9cea097a23cc186f8602265b8e2a6a835adf
3
+ size 1064
checkpoint-30000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b784f83d37150fdafdf102eef381f031c4824c7253feb69d32db800a507bf953
3
+ size 1182072760
checkpoint-30000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c6da8aecbfdb9325bfb0ff4304d53e8b36892f0989e81a9026b2340dc44bac9
3
+ size 185964888
checkpoint-30000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14215f755551527cbb4a12d33823895eb3d261fd24e02d5bffbf27dc81df2376
3
+ size 14408
checkpoint-30000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8b4b0a40f4ca18f123f3c0c1c3c7a0a40f5dc76979d956d08d1ec2c1ff5eee1
3
+ size 1064
image_0.png ADDED

Git LFS Details

  • SHA256: 8349c7643970d201f621a50ce5adde1775264e7e4e91851e968ad1f9ac4baed5
  • Pointer size: 132 Bytes
  • Size of remote file: 1.38 MB
image_1.png ADDED

Git LFS Details

  • SHA256: cbdef1d2f08629f8a5b8b19d7d5ac8b09705d212c653383739753cecb03e4656
  • Pointer size: 132 Bytes
  • Size of remote file: 1.35 MB
image_2.png ADDED

Git LFS Details

  • SHA256: c82957d29cbc22fc98d8fedf4b6866f29359159fe3fd13eb48c032aeba25818d
  • Pointer size: 132 Bytes
  • Size of remote file: 1.38 MB
image_3.png ADDED

Git LFS Details

  • SHA256: 0b7843f306cf7a509b8a9869836fd1646870e8b340ff537b73ac965896287c35
  • Pointer size: 132 Bytes
  • Size of remote file: 1.32 MB
logs/dreambooth-lora-sd-xl/1727905570.2759311/events.out.tfevents.1727905570.50c876563f2c.4524.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:137001dce5850070ddb0a783268e1037e01bf0076f01b9d38e7532551f9c6f4c
3
+ size 3867
logs/dreambooth-lora-sd-xl/1727905570.2780905/hparams.yml ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: null
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ cache_latents: false
9
+ caption_column: prompt
10
+ center_crop: false
11
+ checkpointing_steps: 10000
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: style
15
+ clip_skip: null
16
+ dataloader_num_workers: 0
17
+ dataset_config_name: null
18
+ dataset_name: BruBel/lor-style-training
19
+ do_edm_style_training: false
20
+ enable_xformers_memory_efficient_attention: false
21
+ gradient_accumulation_steps: 1
22
+ gradient_checkpointing: true
23
+ hub_model_id: null
24
+ hub_token: null
25
+ image_column: image
26
+ instance_data_dir: null
27
+ instance_prompt: In <s0><s1> style.
28
+ learning_rate: 0.0001
29
+ local_rank: -1
30
+ logging_dir: logs
31
+ lora_unet_blocks: null
32
+ lr_num_cycles: 1
33
+ lr_power: 1.0
34
+ lr_scheduler: constant
35
+ lr_warmup_steps: 0
36
+ max_grad_norm: 1.0
37
+ max_train_steps: 30000
38
+ mixed_precision: bf16
39
+ noise_offset: 0
40
+ num_class_images: 100
41
+ num_new_tokens_per_abstraction: 2
42
+ num_train_epochs: 2000
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: lor-style-training-v6
46
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
47
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
48
+ prior_generation_precision: null
49
+ prior_loss_weight: 1.0
50
+ prodigy_beta3: null
51
+ prodigy_decouple: true
52
+ prodigy_safeguard_warmup: true
53
+ prodigy_use_bias_correction: true
54
+ push_to_hub: true
55
+ random_flip: false
56
+ rank: 64
57
+ repeats: 1
58
+ report_to: tensorboard
59
+ resolution: 1024
60
+ resume_from_checkpoint: null
61
+ revision: null
62
+ sample_batch_size: 4
63
+ scale_lr: false
64
+ seed: 1
65
+ snr_gamma: 5.0
66
+ text_encoder_lr: 0.0003
67
+ token_abstraction: LEGENDSOFRUNETERRA
68
+ train_batch_size: 7
69
+ train_text_encoder: false
70
+ train_text_encoder_frac: 1.0
71
+ train_text_encoder_ti: true
72
+ train_text_encoder_ti_frac: 0.5
73
+ use_8bit_adam: false
74
+ use_blora: false
75
+ use_dora: false
76
+ validation_epochs: 50
77
+ validation_prompt: A Poro situated in a red sky, war scenario. The Poro wears a brown
78
+ cape. Region of Shurima. In <s0><s1> style.
79
+ variant: null
80
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1727905570.50c876563f2c.4524.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcf510bd1bb8283f90c06e0ba516e3f70e09ff72ced060264b734274a921708e
3
+ size 226151577
lor-style-training-v6.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee6203821e3e2a4ff9dac647c94b8313e67468a698557598b7d06b38e0b2aeaf
3
+ size 371841776
lor-style-training-v6_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8abf4463b21dd6001f0760b8bcefb22058f9e7888c7757a795eaaba80a3f2e27
3
+ size 16536
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc513240edc71312102e1ab2dce0e451b782cc111f1fc018cee2908850a16ae3
3
+ size 371758976