kmilesz commited on
Commit
b6058de
·
verified ·
1 Parent(s): f44d112

Upload folder using huggingface_hub

Browse files
README.md CHANGED
@@ -1,17 +1,92 @@
1
  ---
2
- license: creativeml-openrail-m
3
- base_model: stabilityai/stable-diffusion-2-1-base
4
- instance_prompt: swd
5
  tags:
6
- - stable-diffusion
7
- - stable-diffusion-diffusers
8
  - text-to-image
9
  - diffusers
10
  - lora
11
- inference: true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  ---
13
- # LoRA DreamBooth - swd
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 "swd" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
 
1
  ---
 
 
 
2
  tags:
3
+ - stable-diffusion-xl
4
+ - stable-diffusion-xl-diffusers
5
  - text-to-image
6
  - diffusers
7
  - lora
8
+ - template:sd-lora
9
+ widget:
10
+ - text: A photo of swd
11
+ output:
12
+ url: image-0.png
13
+ - text: A photo of swd
14
+ output:
15
+ url: image-1.png
16
+ - text: A photo of swd
17
+ output:
18
+ url: image-2.png
19
+ - text: A photo of swd
20
+ output:
21
+ url: image-3.png
22
+ - text: A photo of swd
23
+ output:
24
+ url: image-4.png
25
+ - text: A photo of swd
26
+ output:
27
+ url: image-5.png
28
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
29
+ instance_prompt: A photo of kmilesz/swd
30
+ license: openrail++
31
  ---
 
32
 
33
+ # SDXL LoRA DreamBooth - kmilesz/swd
34
 
35
+ <Gallery />
36
+
37
+ ## Model description
38
+
39
+ ### These are kmilesz/swd LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
40
+
41
+ ## Download model
42
+
43
+ ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
44
+
45
+ - **LoRA**: download **[`swd.safetensors` here 💾](/kmilesz/swd/blob/main/swd.safetensors)**.
46
+ - Place it on your `models/Lora` folder.
47
+ - On AUTOMATIC1111, load the LoRA by adding `<lora:swd:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
48
+ - *Embeddings*: download **[`swd_emb.safetensors` here 💾](/kmilesz/swd/blob/main/swd_emb.safetensors)**.
49
+ - Place it on it on your `embeddings` folder
50
+ - Use it by adding `swd_emb` to your prompt. For example, `A photo of swd`
51
+ (you need both the LoRA and the embeddings as they were trained together for this LoRA)
52
+
53
+
54
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
55
+
56
+ ```py
57
+ from diffusers import AutoPipelineForText2Image
58
+ import torch
59
+ from huggingface_hub import hf_hub_download
60
+ from safetensors.torch import load_file
61
+
62
+ pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
63
+ pipeline.load_lora_weights('kmilesz/swd', weight_name='pytorch_lora_weights.safetensors')
64
+ embedding_path = hf_hub_download(repo_id='kmilesz/swd', filename='swd_emb.safetensors' repo_type="model")
65
+ state_dict = load_file(embedding_path)
66
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=[], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
67
+ pipeline.load_textual_inversion(state_dict["clip_g"], token=[], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
68
+
69
+ image = pipeline('A photo of kmilesz/swd').images[0]
70
+ ```
71
+
72
+ 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)
73
+
74
+ ## Trigger words
75
+
76
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
77
+
78
+ to trigger concept `TOK` → use `<s0><s1>` in your prompt
79
+
80
+
81
+
82
+ ## Details
83
+ All [Files & versions](/kmilesz/swd/tree/main).
84
+
85
+ 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).
86
+
87
+ LoRA for the text encoder was enabled. False.
88
+
89
+ Pivotal tuning was enabled: True.
90
+
91
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
92
 
image-0.png ADDED
image-1.png ADDED
image-2.png ADDED
image-3.png ADDED
image-4.png ADDED
image-5.png ADDED
logs/dreambooth-lora-sd-xl/1719377045.4977727/events.out.tfevents.1719377045.r-kmilesz-autotrain-swd-2o1q0jop-f0070-nqwtq.194.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c02ecc801ed1a8c9206e7590cff07498f5276558440d3403a5bac567729be55
3
+ size 3556
logs/dreambooth-lora-sd-xl/1719377045.4991412/hparams.yml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
9
+ caption_column: prompt
10
+ center_crop: false
11
+ checkpointing_steps: 100000
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: null
15
+ crops_coords_top_left_h: 0
16
+ crops_coords_top_left_w: 0
17
+ dataloader_num_workers: 0
18
+ dataset_config_name: null
19
+ dataset_name: ./1a761225-f253-4a37-964b-6a997a44dae8
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: A photo of swd
28
+ learning_rate: 1.0
29
+ local_rank: -1
30
+ logging_dir: logs
31
+ lr_num_cycles: 1
32
+ lr_power: 1.0
33
+ lr_scheduler: constant
34
+ lr_warmup_steps: 0
35
+ max_grad_norm: 1.0
36
+ max_train_steps: 500
37
+ mixed_precision: bf16
38
+ noise_offset: 0
39
+ num_class_images: 100
40
+ num_new_tokens_per_abstraction: 2
41
+ num_train_epochs: 56
42
+ num_validation_images: 4
43
+ optimizer: prodigy
44
+ output_dir: swd
45
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
46
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
47
+ prior_generation_precision: null
48
+ prior_loss_weight: 1.0
49
+ prodigy_beta3: null
50
+ prodigy_decouple: true
51
+ prodigy_safeguard_warmup: true
52
+ prodigy_use_bias_correction: true
53
+ push_to_hub: false
54
+ rank: 16
55
+ repeats: 3
56
+ report_to: tensorboard
57
+ resolution: 1024
58
+ resume_from_checkpoint: null
59
+ revision: null
60
+ sample_batch_size: 4
61
+ scale_lr: false
62
+ seed: 42
63
+ snr_gamma: null
64
+ text_encoder_lr: 1.0
65
+ token_abstraction: TOK
66
+ train_batch_size: 2
67
+ train_text_encoder: false
68
+ train_text_encoder_frac: 1.0
69
+ train_text_encoder_ti: true
70
+ train_text_encoder_ti_frac: 0.5
71
+ use_8bit_adam: false
72
+ validation_epochs: 50
73
+ validation_prompt: null
74
+ variant: null
75
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1719377045.r-kmilesz-autotrain-swd-2o1q0jop-f0070-nqwtq.194.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1c048440864d7a57eee5867b7f491057317a70f14ea82f2bff9897ff709da29
3
+ size 41834
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5721b346062641c65d2bd2f20e8822b7b994d5dfcdca68ae7b36f4805e0d658
3
+ size 93065304
swd.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf6be2c811ed821b0996dc9b3aca5abb7ae0f68f3d6d6e370f7cf343cefbdc42
3
+ size 93148104
swd_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7870e34acf5deff207fb7421e069459ae6664f97fd35945f6155fc70ffe85ce
3
+ size 16536