qd2107 commited on
Commit
f33a54c
·
verified ·
1 Parent(s): 67f4df7

End of training

Browse files
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 water pokemon
6
+ widget:
7
+ - text: a water pokemon
8
+ output:
9
+ url: image_0.png
10
+ - text: a water pokemon
11
+ output:
12
+ url: image_1.png
13
+ - text: a water pokemon
14
+ output:
15
+ url: image_2.png
16
+ - text: a water pokemon
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 - qd2107/water-trained-sd3-lora
34
+
35
+ <Gallery />
36
+
37
+ ## Model description
38
+
39
+ These are qd2107/water-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 water pokemon` to trigger the image generation.
48
+
49
+ ## Download model
50
+
51
+ [Download the *.safetensors LoRA](qd2107/water-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('qd2107/water-trained-sd3-lora', weight_name='pytorch_lora_weights.safetensors')
60
+ image = pipeline('a water pokemon').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 💾](/qd2107/water-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:343891a1d90f53a6707b72c507409885e00b197906e3146f6ffbd0ed56998087
3
+ size 19096716
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d27c5310b320313bc8b7bd19252822d1006ca5418ed13120b810d635d4b4385
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:9d119c329937e57460f46003241e879043f4847e95da87d9486cc246d3f5899b
3
+ size 14344
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18b984273ea2d45b7ffb1d047bb359d93111e41fcad70d16a1b453fd38f72636
3
+ size 988
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68d7b85c46d195e3cd6e587d94878d77f315e0a073acfab811251f39fbe891fc
3
+ size 1000
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/dreambooth-sd3-lora/1734277647.1843097/events.out.tfevents.1734277647.970805768a22.3981.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6656d717d125ab3095aae2967683d2bfac153b4969eb87cf11a3bcff1e5c3ff2
3
+ size 3395
logs/dreambooth-sd3-lora/1734277647.190974/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: /content/drive/MyDrive/Colab Notebooks/pic/Water
24
+ instance_prompt: a water pokemon
25
+ learning_rate: 1.0e-05
26
+ local_rank: -1
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: 16
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: /content/drive/MyDrive/water-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 water pokemon
72
+ variant: null
73
+ weighting_scheme: logit_normal
74
+ with_prior_preservation: false
logs/dreambooth-sd3-lora/events.out.tfevents.1734277647.970805768a22.3981.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:648adec6ea1b7edc7c5d5f7341325c9ecfd928a572860ad73047308f3ddbc11a
3
+ size 7090306
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:381f21d4624cf5e2c6499795b9a048b51dedd14284c28d026237f73a7a8a6e5d
3
+ size 4743176