shawn323 commited on
Commit
ea3fffb
·
verified ·
1 Parent(s): 5a3d7c7

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: in naruto art style
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 - shawn323/flux-lora-naruto
22
+
23
+ <Gallery />
24
+
25
+ ## Model description
26
+
27
+ These are shawn323/flux-lora-naruto 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 `in naruto art style` to trigger the image generation.
36
+
37
+ ## Download model
38
+
39
+ [Download the *.safetensors LoRA](shawn323/flux-lora-naruto/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('shawn323/flux-lora-naruto', weight_name='pytorch_lora_weights.safetensors')
48
+ image = pipeline('in naruto art style').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:507f08b50317fa2092e07083858fb1d041561480d3fc31eade2d9b9925424720
3
+ size 91854485
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c84abee9d5d514f6c2aaa48e64034edc5586b6ab279029323f7f4b9d3a24873c
3
+ size 89747048
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3128faa68e9e810889cf7d94b81b1c8b4bbf7a4abc9817748f611e6cf693af1
3
+ size 14821
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8baa45cc00e6da1eefbbbc2d50fb568432ef1e20241f0a3571b49dcb8d62f7f2
3
+ size 1401
logs/dreambooth-flux-dev-lora/1764717482.0682497/events.out.tfevents.1764717482.807afb8ca9ea.14278.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aed99c8bbf51583d68c3c71da937dd43561441493228d33edea8337542ab8ec9
3
+ size 3471
logs/dreambooth-flux-dev-lora/1764717482.069978/hparams.yml ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ enable_npu_flash_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: true
21
+ guidance_scale: 3.5
22
+ hub_model_id: shawn323/flux-lora-naruto
23
+ hub_token: null
24
+ image_column: image
25
+ instance_data_dir: /content/data/naruto
26
+ instance_prompt: in naruto art style
27
+ learning_rate: 0.0001
28
+ local_rank: -1
29
+ logging_dir: logs
30
+ logit_mean: 0.0
31
+ logit_std: 1.0
32
+ lora_alpha: 4
33
+ lora_dropout: 0.0
34
+ lora_layers: null
35
+ lr_num_cycles: 1
36
+ lr_power: 1.0
37
+ lr_scheduler: constant
38
+ lr_warmup_steps: 0
39
+ max_grad_norm: 1.0
40
+ max_sequence_length: 512
41
+ max_train_steps: 500
42
+ mixed_precision: bf16
43
+ mode_scale: 1.29
44
+ num_class_images: 100
45
+ num_train_epochs: 46
46
+ num_validation_images: 4
47
+ optimizer: AdamW
48
+ output_dir: /content/lora-naruto-flux
49
+ pretrained_model_name_or_path: black-forest-labs/FLUX.1-dev
50
+ prior_generation_precision: null
51
+ prior_loss_weight: 1.0
52
+ prodigy_beta3: null
53
+ prodigy_decouple: true
54
+ prodigy_safeguard_warmup: true
55
+ prodigy_use_bias_correction: true
56
+ push_to_hub: true
57
+ random_flip: false
58
+ rank: 16
59
+ repeats: 1
60
+ report_to: tensorboard
61
+ resolution: 512
62
+ resume_from_checkpoint: null
63
+ revision: null
64
+ sample_batch_size: 4
65
+ scale_lr: false
66
+ seed: 42
67
+ text_encoder_lr: 5.0e-06
68
+ train_batch_size: 4
69
+ train_text_encoder: false
70
+ upcast_before_saving: false
71
+ use_8bit_adam: true
72
+ validation_epochs: 50
73
+ validation_prompt: null
74
+ variant: null
75
+ weighting_scheme: none
76
+ with_prior_preservation: false
logs/dreambooth-flux-dev-lora/events.out.tfevents.1764717482.807afb8ca9ea.14278.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23ba73c21eb4ee27409c4e2e4302eb4ebac4931a49e5c1089e3779fdd1854d7e
3
+ size 41834
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c84abee9d5d514f6c2aaa48e64034edc5586b6ab279029323f7f4b9d3a24873c
3
+ size 89747048