Konghao commited on
Commit
3a77036
·
verified ·
1 Parent(s): 4433915

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,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stabilityai/stable-diffusion-3-medium-diffusers
3
+ library_name: diffusers
4
+ license: other
5
+ instance_prompt: a rendered photo of sks woman
6
+ widget:
7
+ - text: a realistic photo of sks woman
8
+ output:
9
+ url: image_0.png
10
+ - text: a realistic photo of sks woman
11
+ output:
12
+ url: image_1.png
13
+ - text: a realistic photo of sks woman
14
+ output:
15
+ url: image_2.png
16
+ - text: a realistic photo of sks woman
17
+ output:
18
+ url: image_3.png
19
+ tags:
20
+ - text-to-image
21
+ - diffusers-training
22
+ - diffusers
23
+ - lora
24
+ - template:sd-lora
25
+ - sd3
26
+ - sd3-diffusers
27
+ ---
28
+
29
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
30
+ should probably proofread and complete it, then remove this comment. -->
31
+
32
+
33
+ # SD3 DreamBooth LoRA - Konghao/trained-sd3-lora
34
+
35
+ <Gallery />
36
+
37
+ ## Model description
38
+
39
+ These are Konghao/trained-sd3-lora DreamBooth LoRA weights for stabilityai/stable-diffusion-3-medium-diffusers.
40
+
41
+ The weights were trained using [DreamBooth](https://dreambooth.github.io/) with the [SD3 diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/README_sd3.md).
42
+
43
+ Was LoRA for the text encoder enabled? False.
44
+
45
+ ## Trigger words
46
+
47
+ You should use `a rendered photo of sks woman` to trigger the image generation.
48
+
49
+ ## Download model
50
+
51
+ [Download the *.safetensors LoRA](Konghao/trained-sd3-lora/tree/main) in the Files & versions tab.
52
+
53
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
54
+
55
+ ```py
56
+ from diffusers import AutoPipelineForText2Image
57
+ import torch
58
+ pipeline = AutoPipelineForText2Image.from_pretrained(stabilityai/stable-diffusion-3-medium-diffusers, torch_dtype=torch.float16).to('cuda')
59
+ pipeline.load_lora_weights('Konghao/trained-sd3-lora', weight_name='pytorch_lora_weights.safetensors')
60
+ image = pipeline('a realistic photo of sks woman').images[0]
61
+ ```
62
+
63
+ ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
64
+
65
+ - **LoRA**: download **[`diffusers_lora_weights.safetensors` here 💾](/Konghao/trained-sd3-lora/blob/main/diffusers_lora_weights.safetensors)**.
66
+ - Rename it and place it on your `models/Lora` folder.
67
+ - On AUTOMATIC1111, load the LoRA by adding `<lora:your_new_name:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
68
+
69
+ 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)
70
+
71
+ ## License
72
+
73
+ Please adhere to the licensing terms as described [here](https://huggingface.co/stabilityai/stable-diffusion-3-medium/blob/main/LICENSE.md).
74
+
75
+
76
+ ## Intended uses & limitations
77
+
78
+ #### How to use
79
+
80
+ ```python
81
+ # TODO: add an example code snippet for running this diffusion pipeline
82
+ ```
83
+
84
+ #### Limitations and bias
85
+
86
+ [TODO: provide examples of latent issues and potential remediations]
87
+
88
+ ## Training details
89
+
90
+ [TODO: describe the data used to train the model]
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abaa705b7479fc019b993a3dbed2dab0f1421ae2a2fb704a9767858fb4b15ecd
3
+ size 19097181
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b363b88f783ad67e3517a37cee8482a3103393da36e69a2f0c155ed769134e7d
3
+ size 9437288
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd3f7844247a2ca0134c1eb39763588175ba015641b2c817d3ea1aa0c7372ed2
3
+ size 16449
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f77569c2e850b04af982cc8c1389f1430851448915c593b69e5da36ce05b71d7
3
+ size 1383
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3bb6a92a1dfebbc2f0a3dd1b7ba911e9665bdfd9b3136756090a9b6ccd3aaec0
3
+ size 1401
image_0.png ADDED

Git LFS Details

  • SHA256: a2b4844949063deb1f9d5afb81125123398bbe07e7e7c46dfa417c6e4726bc5e
  • Pointer size: 131 Bytes
  • Size of remote file: 841 kB
image_1.png ADDED

Git LFS Details

  • SHA256: 3836c8f30e2eb7ecb0a44309f018d9e910a6a82ccc607b130c99b2c4d6e57c6b
  • Pointer size: 132 Bytes
  • Size of remote file: 1 MB
image_2.png ADDED

Git LFS Details

  • SHA256: 3c9131395715da62128db8d04be1e93465e7061b76907a476672102d7984e842
  • Pointer size: 131 Bytes
  • Size of remote file: 770 kB
image_3.png ADDED

Git LFS Details

  • SHA256: 7f98116695dde90b93cecaf4b5076d75f8bc3f90e8ae1948fba97f60117714a4
  • Pointer size: 132 Bytes
  • Size of remote file: 1.01 MB
logs/dreambooth-sd3-lora/1761349185.7579193/events.out.tfevents.1761349185.phe108c-ruishanl-02.425621.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e966f0c13efeaa56dac9fa12a6076cd164a6944358814e0cc21b4312bd30b9b2
3
+ size 3320
logs/dreambooth-sd3-lora/1761349185.7597952/hparams.yml ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ cache_latents: false
9
+ caption_column: null
10
+ center_crop: false
11
+ checkpointing_steps: 500
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: null
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: null
18
+ gradient_accumulation_steps: 4
19
+ gradient_checkpointing: false
20
+ hub_model_id: null
21
+ hub_token: null
22
+ image_column: image
23
+ instance_data_dir: /home/konghaoz/diffusers/facetime_data/batch_0/
24
+ instance_prompt: a rendered photo of sks woman
25
+ learning_rate: 0.0004
26
+ local_rank: 0
27
+ logging_dir: logs
28
+ logit_mean: 0.0
29
+ logit_std: 1.0
30
+ lora_blocks: null
31
+ lora_layers: null
32
+ lr_num_cycles: 1
33
+ lr_power: 1.0
34
+ lr_scheduler: constant
35
+ lr_warmup_steps: 500
36
+ max_grad_norm: 1.0
37
+ max_sequence_length: 77
38
+ max_train_steps: 63
39
+ mixed_precision: fp16
40
+ mode_scale: 1.29
41
+ num_class_images: 100
42
+ num_train_epochs: 1
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: /home/konghaoz/diffusers/results/trained-sd3-lora
46
+ precondition_outputs: 1
47
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-3-medium-diffusers
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: false
55
+ random_flip: false
56
+ rank: 4
57
+ repeats: 1
58
+ report_to: tensorboard
59
+ resolution: 512
60
+ resume_from_checkpoint: null
61
+ revision: null
62
+ sample_batch_size: 4
63
+ scale_lr: false
64
+ seed: null
65
+ text_encoder_lr: 5.0e-06
66
+ train_batch_size: 1
67
+ train_text_encoder: false
68
+ upcast_before_saving: false
69
+ use_8bit_adam: false
70
+ validation_epochs: 50
71
+ validation_prompt: null
72
+ variant: null
73
+ weighting_scheme: logit_normal
74
+ with_prior_preservation: false
logs/dreambooth-sd3-lora/1761350300.7736049/events.out.tfevents.1761350300.phe108c-ruishanl-02.449405.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03449221ea68434c83756099bcf5b3c42a3ffb47d9d87333e59ecd71cbcb8e5f
3
+ size 3427
logs/dreambooth-sd3-lora/1761350300.7754164/hparams.yml ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ cache_latents: false
9
+ caption_column: null
10
+ center_crop: false
11
+ checkpointing_steps: 500
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: null
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: null
18
+ gradient_accumulation_steps: 4
19
+ gradient_checkpointing: false
20
+ hub_model_id: null
21
+ hub_token: null
22
+ image_column: image
23
+ instance_data_dir: /home/konghaoz/diffusers/facetime_data/batch_0/
24
+ instance_prompt: a rendered photo of sks woman
25
+ learning_rate: 0.0004
26
+ local_rank: 0
27
+ logging_dir: logs
28
+ logit_mean: 0.0
29
+ logit_std: 1.0
30
+ lora_blocks: null
31
+ lora_layers: 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_sequence_length: 77
38
+ max_train_steps: 500
39
+ mixed_precision: fp16
40
+ mode_scale: 1.29
41
+ num_class_images: 100
42
+ num_train_epochs: 8
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: /home/konghaoz/diffusers/results/trained-sd3-lora
46
+ precondition_outputs: 1
47
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-3-medium-diffusers
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: 4
57
+ repeats: 1
58
+ report_to: tensorboard
59
+ resolution: 512
60
+ resume_from_checkpoint: null
61
+ revision: null
62
+ sample_batch_size: 4
63
+ scale_lr: false
64
+ seed: 0
65
+ text_encoder_lr: 5.0e-06
66
+ train_batch_size: 1
67
+ train_text_encoder: false
68
+ upcast_before_saving: false
69
+ use_8bit_adam: false
70
+ validation_epochs: 25
71
+ validation_prompt: a realistic photo of sks woman
72
+ variant: null
73
+ weighting_scheme: logit_normal
74
+ with_prior_preservation: false
logs/dreambooth-sd3-lora/events.out.tfevents.1761349185.phe108c-ruishanl-02.425621.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4b1f1494c87eb25431a0335bbc116922c17dc16e1b181783ef09abcad569ff6
3
+ size 20658
logs/dreambooth-sd3-lora/events.out.tfevents.1761350300.phe108c-ruishanl-02.449405.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f161ab06e966ce52850b369ec633d8f5dc99e07335a4aa0d5af90752bd8cd093
3
+ size 8250425
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:517d2c259009bf9678749ee7c400e8217b87a9b9dac3b11dfc6b94e3d1ed06f2
3
+ size 4743176