End of training

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+image_0.png filter=lfs diff=lfs merge=lfs -text
+image_1.png filter=lfs diff=lfs merge=lfs -text
+image_2.png filter=lfs diff=lfs merge=lfs -text
+image_3.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,84 @@
+---
+base_model: black-forest-labs/FLUX.1-dev
+library_name: diffusers
+license: other
+instance_prompt: a photo of sks dog
+widget:
+- text: A photo of sks dog in a bucket
+  output:
+    url: image_0.png
+- text: A photo of sks dog in a bucket
+  output:
+    url: image_1.png
+- text: A photo of sks dog in a bucket
+  output:
+    url: image_2.png
+- text: A photo of sks dog in a bucket
+  output:
+    url: image_3.png
+tags:
+- text-to-image
+- diffusers-training
+- diffusers
+- lora
+- flux
+- flux-diffusers
+- template:sd-lora
+---
+
+<!-- This model card has been generated automatically according to the information the training script had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+
+# Flux DreamBooth LoRA - xgemstarx/trained-flux
+
+<Gallery />
+
+## Model description
+
+These are xgemstarx/trained-flux DreamBooth LoRA weights for black-forest-labs/FLUX.1-dev.
+
+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).
+
+Was LoRA for the text encoder enabled? False.
+
+## Trigger words
+
+You should use `a photo of sks dog` to trigger the image generation.
+
+## Download model
+
+[Download the *.safetensors LoRA](xgemstarx/trained-flux/tree/main) in the Files & versions tab.
+
+## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
+
+```py
+from diffusers import AutoPipelineForText2Image
+import torch
+pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
+pipeline.load_lora_weights('xgemstarx/trained-flux', weight_name='pytorch_lora_weights.safetensors')
+image = pipeline('A photo of sks dog in a bucket').images[0]
+```
+
+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)
+
+## License
+
+Please adhere to the licensing terms as described [here](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md).
+
+
+## Intended uses & limitations
+
+#### How to use
+
+```python
+# TODO: add an example code snippet for running this diffusion pipeline
+```
+
+#### Limitations and bias
+
+[TODO: provide examples of latent issues and potential remediations]
+
+## Training details
+
+[TODO: describe the data used to train the model]

checkpoint-500/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1d4a0a0d23bb689549d1b8c04faf68f6ecface9b8bfb5fd64162a39aaf83fd0
+size 79235795

checkpoint-500/pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21ae4f9a9fea5faef6895db4071949ca4e9603c20a6be96b6c0255b396cd624b
+size 22505872

checkpoint-500/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea4056fa4313cdb95ac012a8e8edb767d6af0cc289831b852f7918900fe70ea1
+size 14757

checkpoint-500/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d3900682bea1b95a221ce2ff178a9d6e10581809a9e5aa1ae8a8560d6a3c3df3
+size 1401

logs/dreambooth-flux-dev-lora/1769646058.7533526/events.out.tfevents.1769646058.192-222-52-137.355678.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a48b90240a5d5ef5265c3c644e2a19d681d143957c3a0f28040f3881e464578c
+size 3303

logs/dreambooth-flux-dev-lora/1769646058.7545862/hparams.yml

@@ -0,0 +1,71 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+adam_weight_decay_text_encoder: 0.001
+allow_tf32: false
+cache_dir: null
+caption_column: null
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_npu_flash_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+guidance_scale: 1.0
+hub_model_id: null
+hub_token: null
+image_column: image
+image_interpolation_mode: lanczos
+instance_data_dir: dog
+instance_prompt: a photo of sks dog
+learning_rate: 1.0
+local_rank: -1
+logging_dir: logs
+logit_mean: 0.0
+logit_std: 1.0
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_sequence_length: 77
+max_train_steps: 500
+mixed_precision: bf16
+mode_scale: 1.29
+num_class_images: 100
+num_train_epochs: 250
+num_validation_images: 4
+optimizer: prodigy
+output_dir: trained-flux
+pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
+prior_generation_precision: null
+prior_loss_weight: 1.0
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+push_to_hub: true
+random_flip: false
+repeats: 1
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+text_encoder_lr: 5.0e-06
+train_batch_size: 1
+train_text_encoder: false
+use_8bit_adam: false
+validation_epochs: 25
+validation_prompt: A photo of sks dog in a bucket
+variant: null
+weighting_scheme: none
+with_prior_preservation: false

logs/dreambooth-flux-dev-lora/1769646147.5781498/events.out.tfevents.1769646147.192-222-52-137.355811.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f8971a4c675acb7b2a82556369cb61bdb669e34669d1be6de3040e01adc4a988
+size 3303

logs/dreambooth-flux-dev-lora/1769646147.5793717/hparams.yml

@@ -0,0 +1,71 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+adam_weight_decay_text_encoder: 0.001
+allow_tf32: false
+cache_dir: null
+caption_column: null
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_npu_flash_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+guidance_scale: 1.0
+hub_model_id: null
+hub_token: null
+image_column: image
+image_interpolation_mode: lanczos
+instance_data_dir: dog
+instance_prompt: a photo of sks dog
+learning_rate: 1.0
+local_rank: -1
+logging_dir: logs
+logit_mean: 0.0
+logit_std: 1.0
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_sequence_length: 77
+max_train_steps: 500
+mixed_precision: bf16
+mode_scale: 1.29
+num_class_images: 100
+num_train_epochs: 250
+num_validation_images: 4
+optimizer: prodigy
+output_dir: trained-flux
+pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
+prior_generation_precision: null
+prior_loss_weight: 1.0
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+push_to_hub: true
+random_flip: false
+repeats: 1
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+text_encoder_lr: 5.0e-06
+train_batch_size: 1
+train_text_encoder: false
+use_8bit_adam: false
+validation_epochs: 25
+validation_prompt: A photo of sks dog in a bucket
+variant: null
+weighting_scheme: none
+with_prior_preservation: false

logs/dreambooth-flux-dev-lora/1769647068.4661674/events.out.tfevents.1769647068.192-222-52-137.356186.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6da26279233773418d331e6e1b70b7fa56a4d2751563103973c1d6dcbfc3b8c4
+size 3303

logs/dreambooth-flux-dev-lora/1769647068.4675112/hparams.yml

@@ -0,0 +1,71 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+adam_weight_decay_text_encoder: 0.001
+allow_tf32: false
+cache_dir: null
+caption_column: null
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_npu_flash_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+guidance_scale: 1.0
+hub_model_id: null
+hub_token: null
+image_column: image
+image_interpolation_mode: lanczos
+instance_data_dir: dog
+instance_prompt: a photo of sks dog
+learning_rate: 1.0
+local_rank: -1
+logging_dir: logs
+logit_mean: 0.0
+logit_std: 1.0
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_sequence_length: 77
+max_train_steps: 500
+mixed_precision: bf16
+mode_scale: 1.29
+num_class_images: 100
+num_train_epochs: 250
+num_validation_images: 4
+optimizer: prodigy
+output_dir: trained-flux
+pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
+prior_generation_precision: null
+prior_loss_weight: 1.0
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+push_to_hub: true
+random_flip: false
+repeats: 1
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+text_encoder_lr: 5.0e-06
+train_batch_size: 1
+train_text_encoder: false
+use_8bit_adam: false
+validation_epochs: 25
+validation_prompt: A photo of sks dog in a bucket
+variant: null
+weighting_scheme: none
+with_prior_preservation: false

logs/dreambooth-flux-dev-lora/1769647275.205348/events.out.tfevents.1769647275.192-222-52-137.356301.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3b6061f384b618b38657f289819ecdf1cfd30995b185e716eb49457c4a85590
+size 3457

logs/dreambooth-flux-dev-lora/1769647275.2065892/hparams.yml

@@ -0,0 +1,76 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+adam_weight_decay_text_encoder: 0.001
+allow_tf32: false
+cache_dir: null
+cache_latents: false
+caption_column: null
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: null
+enable_npu_flash_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+guidance_scale: 1.0
+hub_model_id: null
+hub_token: null
+image_column: image
+instance_data_dir: dog
+instance_prompt: a photo of sks dog
+learning_rate: 1.0
+local_rank: -1
+logging_dir: logs
+logit_mean: 0.0
+logit_std: 1.0
+lora_alpha: 4
+lora_dropout: 0.0
+lora_layers: null
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_sequence_length: 512
+max_train_steps: 500
+mixed_precision: bf16
+mode_scale: 1.29
+num_class_images: 100
+num_train_epochs: 250
+num_validation_images: 4
+optimizer: prodigy
+output_dir: trained-flux
+pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
+prior_generation_precision: null
+prior_loss_weight: 1.0
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+push_to_hub: true
+random_flip: false
+rank: 4
+repeats: 1
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+text_encoder_lr: 5.0e-06
+train_batch_size: 1
+train_text_encoder: false
+upcast_before_saving: false
+use_8bit_adam: false
+validation_epochs: 25
+validation_prompt: A photo of sks dog in a bucket
+variant: null
+weighting_scheme: none
+with_prior_preservation: false

logs/dreambooth-flux-dev-lora/events.out.tfevents.1769646058.192-222-52-137.355678.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e70de3b4149464e6bac92282d4544bb152d720f948a4cecabc4c778126c29459
+size 88

logs/dreambooth-flux-dev-lora/events.out.tfevents.1769646147.192-222-52-137.355811.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4bb1758c3141d94a4b61f49d27822e05dccba2271d67e1eb6fc0b9a7b6081e3
+size 88

logs/dreambooth-flux-dev-lora/events.out.tfevents.1769647068.192-222-52-137.356186.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1301c2c8a3c40fe2bdf170a9d6614e25dc79332a8b0e9d353cc7ebe8dabdc5e4
+size 88

logs/dreambooth-flux-dev-lora/events.out.tfevents.1769647275.192-222-52-137.356301.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3d293feedb0dd5f81c4c901ef4802f1caf4a981d10ebdd8f48b20d9cd9be895
+size 58890083

pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21ae4f9a9fea5faef6895db4071949ca4e9603c20a6be96b6c0255b396cd624b
+size 22505872