Pikassik commited on
Commit
e9b9167
·
verified ·
1 Parent(s): a944678

End of training

Browse files
README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: black-forest-labs/FLUX.1-dev
3
+ library_name: diffusers
4
+ license: other
5
+ instance_prompt: a photo of sks dog
6
+ widget: []
7
+ tags:
8
+ - text-to-image
9
+ - diffusers-training
10
+ - diffusers
11
+ - lora
12
+ - flux
13
+ - flux-diffusers
14
+ - template:sd-lora
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+
21
+ # Flux DreamBooth LoRA - Pikassik/save_model
22
+
23
+ <Gallery />
24
+
25
+ ## Model description
26
+
27
+ These are Pikassik/save_model DreamBooth LoRA weights for black-forest-labs/FLUX.1-dev.
28
+
29
+ The weights were trained using [DreamBooth](https://dreambooth.github.io/) with the [Flux diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/README_flux.md).
30
+
31
+ Was LoRA for the text encoder enabled? False.
32
+
33
+ ## Trigger words
34
+
35
+ You should use `a photo of sks dog` to trigger the image generation.
36
+
37
+ ## Download model
38
+
39
+ [Download the *.safetensors LoRA](Pikassik/save_model/tree/main) in the Files & versions tab.
40
+
41
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
42
+
43
+ ```py
44
+ from diffusers import AutoPipelineForText2Image
45
+ import torch
46
+ pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
47
+ pipeline.load_lora_weights('Pikassik/save_model', weight_name='pytorch_lora_weights.safetensors')
48
+ image = pipeline('a photo of sks dog').images[0]
49
+ ```
50
+
51
+ 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)
52
+
53
+ ## License
54
+
55
+ Please adhere to the licensing terms as described [here](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md).
56
+
57
+
58
+ ## Intended uses & limitations
59
+
60
+ #### How to use
61
+
62
+ ```python
63
+ # TODO: add an example code snippet for running this diffusion pipeline
64
+ ```
65
+
66
+ #### Limitations and bias
67
+
68
+ [TODO: provide examples of latent issues and potential remediations]
69
+
70
+ ## Training details
71
+
72
+ [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:63caca747500a41ee5539562cef8841a06d32af97e4adf2ddc01b0c66bc06a47
3
+ size 45402518
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:629eb04bf8235f9f95430ee5babaf33d3b67dbc0b992ea79d8b6eadb7f93e919
3
+ size 22504080
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a8cf82b7b9552dde8cebfa89ae9476879ee488d118eb41b597da71afa9f09cd
3
+ size 14344
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65115106fbd9463fa6c7133b543dece71c31fcba363ff15b8c73f8c7eb0739c5
3
+ size 1000
logs/dreambooth-flux-dev-lora/1741864684.042618/events.out.tfevents.1741864684.paramonov-test-0.45696.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10b1ed93b02e239bcd80314db6b5aba82bb7a86967045540f5d843e99c5e77ce
3
+ size 3245
logs/dreambooth-flux-dev-lora/1741864684.0480893/hparams.yml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: class_images
14
+ class_prompt: a photo of dog
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: null
18
+ gradient_accumulation_steps: 1
19
+ gradient_checkpointing: false
20
+ guidance_scale: 3.5
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: dog
25
+ instance_prompt: a photo of sks dog
26
+ learning_rate: 5.0e-06
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ logit_mean: 0.0
30
+ logit_std: 1.0
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: 512
38
+ max_train_steps: 800
39
+ mixed_precision: null
40
+ mode_scale: 1.29
41
+ num_class_images: 200
42
+ num_train_epochs: 4
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: save_model
46
+ pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
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: true
54
+ random_flip: false
55
+ rank: 4
56
+ repeats: 1
57
+ report_to: tensorboard
58
+ resolution: 512
59
+ resume_from_checkpoint: null
60
+ revision: null
61
+ sample_batch_size: 4
62
+ scale_lr: false
63
+ seed: null
64
+ text_encoder_lr: 5.0e-06
65
+ train_batch_size: 1
66
+ train_text_encoder: false
67
+ upcast_before_saving: false
68
+ use_8bit_adam: false
69
+ validation_epochs: 50
70
+ validation_prompt: null
71
+ variant: null
72
+ weighting_scheme: none
73
+ with_prior_preservation: true
logs/dreambooth-flux-dev-lora/1741865876.96873/events.out.tfevents.1741865876.paramonov-test-0.45867.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5faa70983bc08ce55ba8bee14514f8f76c493af62706140ee45973b7aba8122f
3
+ size 3245
logs/dreambooth-flux-dev-lora/1741865876.9741232/hparams.yml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: class_images
14
+ class_prompt: a photo of dog
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: null
18
+ gradient_accumulation_steps: 1
19
+ gradient_checkpointing: false
20
+ guidance_scale: 3.5
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: dog
25
+ instance_prompt: a photo of sks dog
26
+ learning_rate: 5.0e-06
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ logit_mean: 0.0
30
+ logit_std: 1.0
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: 512
38
+ max_train_steps: 800
39
+ mixed_precision: null
40
+ mode_scale: 1.29
41
+ num_class_images: 200
42
+ num_train_epochs: 4
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: save_model
46
+ pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
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: true
54
+ random_flip: false
55
+ rank: 4
56
+ repeats: 1
57
+ report_to: tensorboard
58
+ resolution: 512
59
+ resume_from_checkpoint: null
60
+ revision: null
61
+ sample_batch_size: 4
62
+ scale_lr: false
63
+ seed: null
64
+ text_encoder_lr: 5.0e-06
65
+ train_batch_size: 1
66
+ train_text_encoder: false
67
+ upcast_before_saving: false
68
+ use_8bit_adam: false
69
+ validation_epochs: 50
70
+ validation_prompt: null
71
+ variant: null
72
+ weighting_scheme: none
73
+ with_prior_preservation: true
logs/dreambooth-flux-dev-lora/1741866081.525809/events.out.tfevents.1741866081.paramonov-test-0.46081.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3faa527492664c5a73903ca92e4ff775b4b392475970ebaea890e62f0ac2c63b
3
+ size 3245
logs/dreambooth-flux-dev-lora/1741866081.533404/hparams.yml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: class_images
14
+ class_prompt: a photo of dog
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: null
18
+ gradient_accumulation_steps: 1
19
+ gradient_checkpointing: false
20
+ guidance_scale: 3.5
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: dog
25
+ instance_prompt: a photo of sks dog
26
+ learning_rate: 5.0e-06
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ logit_mean: 0.0
30
+ logit_std: 1.0
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: 512
38
+ max_train_steps: 800
39
+ mixed_precision: null
40
+ mode_scale: 1.29
41
+ num_class_images: 200
42
+ num_train_epochs: 4
43
+ num_validation_images: 4
44
+ optimizer: AdamW
45
+ output_dir: save_model
46
+ pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
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: true
54
+ random_flip: false
55
+ rank: 4
56
+ repeats: 1
57
+ report_to: tensorboard
58
+ resolution: 512
59
+ resume_from_checkpoint: null
60
+ revision: null
61
+ sample_batch_size: 4
62
+ scale_lr: false
63
+ seed: null
64
+ text_encoder_lr: 5.0e-06
65
+ train_batch_size: 1
66
+ train_text_encoder: false
67
+ upcast_before_saving: false
68
+ use_8bit_adam: false
69
+ validation_epochs: 50
70
+ validation_prompt: null
71
+ variant: null
72
+ weighting_scheme: none
73
+ with_prior_preservation: true
logs/dreambooth-flux-dev-lora/events.out.tfevents.1741864684.paramonov-test-0.45696.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf0bd02700934ce6bcba9a72095d8ab4d3e968b769cbce8a4ed90cee50b98848
3
+ size 88
logs/dreambooth-flux-dev-lora/events.out.tfevents.1741865876.paramonov-test-0.45867.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d910513201f4894a2c671b52ecd1311b765e0a7d2d0826a4f4b40928b40b49b
3
+ size 14618
logs/dreambooth-flux-dev-lora/events.out.tfevents.1741866081.paramonov-test-0.46081.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ddc301a3ef65fed538af8dbfd01f33ddce888c82167bc4afab59f9a45e3d947
3
+ size 67034
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64f95daffe76aa0770b0ebbd48135877db431a291a0351ccfb2fd26be7d4a213
3
+ size 22504080