xgemstarx commited on
Commit
d01d13e
·
verified ·
1 Parent(s): 43d325b

End of training

Browse files
Files changed (33) hide show
  1. README.md +72 -0
  2. checkpoint-1000/optimizer.bin +3 -0
  3. checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
  4. checkpoint-1000/random_states_0.pkl +3 -0
  5. checkpoint-1000/scheduler.bin +3 -0
  6. checkpoint-1500/optimizer.bin +3 -0
  7. checkpoint-1500/pytorch_lora_weights.safetensors +3 -0
  8. checkpoint-1500/random_states_0.pkl +3 -0
  9. checkpoint-1500/scheduler.bin +3 -0
  10. checkpoint-2000/optimizer.bin +3 -0
  11. checkpoint-2000/pytorch_lora_weights.safetensors +3 -0
  12. checkpoint-2000/random_states_0.pkl +3 -0
  13. checkpoint-2000/scheduler.bin +3 -0
  14. checkpoint-2500/optimizer.bin +3 -0
  15. checkpoint-2500/pytorch_lora_weights.safetensors +3 -0
  16. checkpoint-2500/random_states_0.pkl +3 -0
  17. checkpoint-2500/scheduler.bin +3 -0
  18. checkpoint-3000/optimizer.bin +3 -0
  19. checkpoint-3000/pytorch_lora_weights.safetensors +3 -0
  20. checkpoint-3000/random_states_0.pkl +3 -0
  21. checkpoint-3000/scheduler.bin +3 -0
  22. checkpoint-3500/optimizer.bin +3 -0
  23. checkpoint-3500/pytorch_lora_weights.safetensors +3 -0
  24. checkpoint-3500/random_states_0.pkl +3 -0
  25. checkpoint-3500/scheduler.bin +3 -0
  26. checkpoint-500/optimizer.bin +3 -0
  27. checkpoint-500/pytorch_lora_weights.safetensors +3 -0
  28. checkpoint-500/random_states_0.pkl +3 -0
  29. checkpoint-500/scheduler.bin +3 -0
  30. logs/dreambooth-flux-dev-lora/1737427873.7207403/events.out.tfevents.1737427873.192-222-58-137.50555.1 +3 -0
  31. logs/dreambooth-flux-dev-lora/1737427873.729327/hparams.yml +73 -0
  32. logs/dreambooth-flux-dev-lora/events.out.tfevents.1737427873.192-222-58-137.50555.0 +3 -0
  33. pytorch_lora_weights.safetensors +3 -0
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 xjiminx
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 - xgemstarx/smol_model2
22
+
23
+ <Gallery />
24
+
25
+ ## Model description
26
+
27
+ These are xgemstarx/smol_model2 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 xjiminx` to trigger the image generation.
36
+
37
+ ## Download model
38
+
39
+ [Download the *.safetensors LoRA](xgemstarx/smol_model2/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('xgemstarx/smol_model2', weight_name='pytorch_lora_weights.safetensors')
48
+ image = pipeline('a photo of xjiminx').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-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4c035cd7d36e6bb1678bfccef167b3ff8bdf9a604ed3110eb67d0d4cec8d83c
3
+ size 79235394
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed600cd15519f451d6fb3243215eefcb52f2fde94e37ca1f572caa356fbf3994
3
+ size 22504080
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09aa81281a5a3317397cdc226b9c5df582cf261ff7e0777fbbdf2f388ce79d3a
3
+ size 14344
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e55c075b40ce3ab6db2c27ecf3b4c90bc4e612fff3a226e651c44881d6a2404
3
+ size 1000
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82389aff184a31e1fe473f5c944b7738401a990570136415cdbc0ce3fce9a419
3
+ size 79235394
checkpoint-1500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a943c816ff4aba0be4b1d3139c647bc6a7ac56c735a085be03aa09328eaaca21
3
+ size 22504080
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cf5dba66f9d781e98143d743fa58725ffb03eb2cb7599e91b7cf4b4ee5eada2
3
+ size 14344
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab46152b551fb24cef10d97aece42dc7616a1a1b456640ac3f526f7abac6e0e3
3
+ size 1000
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90a5b386d9b83363e29dad326e54b64ca44c9dc63899ae0751dbbd256d2b5756
3
+ size 79235394
checkpoint-2000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d04de370f5a742f899cac642f9a9315208f6de171caed403e9d8af2b1c4bd3a
3
+ size 22504080
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e9f3e8941071cd78c9536636ccdb3e298f2e6fea4f613b1b920498650825902
3
+ size 14344
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:493642aa6dcb7f188f018b7691aba9287429fa60e412f5f63a1430602afc2557
3
+ size 1000
checkpoint-2500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f74fad4bf44011ffccad99f00fc58960d70677db7cdc59ae13097447108b46c
3
+ size 79235394
checkpoint-2500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b32750526743047c4b23ddaa4a24549d36f9704347bb6d1e594ad24d3af93fac
3
+ size 22504080
checkpoint-2500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d748e4f28098a55e17d20ffb8c58c1bc3e1462f9f46ceee3e04b1005fcebe85f
3
+ size 14344
checkpoint-2500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c6b8457d8ed1c4a1141131dfb2267827bfa81f1b7214dc7a16e697888e0382e
3
+ size 1000
checkpoint-3000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:444f315b9d5b9d928e28bf735b48a92205e52f86c83ddcf9b59391f35330807e
3
+ size 79235394
checkpoint-3000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9578d6c5bbcc684b27c454e610e5ead36a228e2c866a526adae75865b2cc9e30
3
+ size 22504080
checkpoint-3000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e054922a152cb5e56e24dfab94fa68c26bcb98086006a72d43ad76e2457ef831
3
+ size 14344
checkpoint-3000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f64a65ba99578a884b4851fbf9e08190ff85a9f85bd177a7358113febdc91ff8
3
+ size 1000
checkpoint-3500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:907b2b7a7cd35d592bd16a39123ef59ed9b63fb083cd90d77d80f87021e4bb63
3
+ size 79235394
checkpoint-3500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58a70728e2f19c018764e512d33a679c3011c2ef5d2728e4a3a6b02ac06d1214
3
+ size 22504080
checkpoint-3500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:858c0bf35bb8539d5362c09ae557144ad5d84006a95597e8e77e32d237a30c51
3
+ size 14344
checkpoint-3500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adc552d20c458f353da0a8721d9dd31f803a69d5ff195a78140771c51334d243
3
+ size 1000
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97f1cf0ef740e1e25e61da28c6e33903605e1e9881832c5b0fbe3863bd740458
3
+ size 79235394
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2498fa691d2972e4d524c1741048ead3a75e791cbd22c014c3d3e65069b4a306
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:c86285e1c45e63e9cb8be9004021c0ffa54bd7e2fbd0aae724b80f14fe4a3544
3
+ size 14344
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d64e43d780f79b2c41c77b3638f3da55c2a3273c6e72a559cfe6fb2b05479b63
3
+ size 1000
logs/dreambooth-flux-dev-lora/1737427873.7207403/events.out.tfevents.1737427873.192-222-58-137.50555.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ba8f8d1077cdbff3487fb1f55bbfd59bed9ac52df6f4b9550f8645252f05f74
3
+ size 3220
logs/dreambooth-flux-dev-lora/1737427873.729327/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: 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
+ guidance_scale: 1.0
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: smol
25
+ instance_prompt: a photo of xjiminx
26
+ learning_rate: 1.0
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: 3500
39
+ mixed_precision: bf16
40
+ mode_scale: 1.29
41
+ num_class_images: 100
42
+ num_train_epochs: 350
43
+ num_validation_images: 4
44
+ optimizer: prodigy
45
+ output_dir: smol_model2
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: 0
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: false
logs/dreambooth-flux-dev-lora/events.out.tfevents.1737427873.192-222-58-137.50555.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b5d1d01bf38da0d278701ac7a2a6eea941ce3ad391e270664db8114e88fb71f
3
+ size 1175054
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58a70728e2f19c018764e512d33a679c3011c2ef5d2728e4a3a6b02ac06d1214
3
+ size 22504080