cwhuh commited on
Commit
5094aa5
·
verified ·
1 Parent(s): 4406f22

End of training

Browse files
README.md ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: black-forest-labs/FLUX.1-dev
3
+ library_name: diffusers
4
+ license: other
5
+ instance_prompt: A newborn <s0><s1><s2><s3><s4> baby.
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 - cwhuh/babyface_flux_dlora_3_Vietnamese
22
+
23
+ <Gallery />
24
+
25
+ ## Model description
26
+
27
+ These are cwhuh/babyface_flux_dlora_3_Vietnamese 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
+ Pivotal tuning was enabled: True.
34
+
35
+ ## Trigger words
36
+
37
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
38
+
39
+ to trigger concept `Vietnamese` → use `<s0><s1><s2><s3><s4>` in your prompt
40
+
41
+
42
+
43
+ ## Download model
44
+
45
+ [Download the *.safetensors LoRA](cwhuh/babyface_flux_dlora_3_Vietnamese/tree/main) in the Files & versions tab.
46
+
47
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
48
+
49
+ ```py
50
+ from diffusers import AutoPipelineForText2Image
51
+ import torch
52
+ from huggingface_hub import hf_hub_download
53
+ from safetensors.torch import load_file
54
+
55
+ pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
56
+ pipeline.load_lora_weights('cwhuh/babyface_flux_dlora_3_Vietnamese', weight_name='pytorch_lora_weights.safetensors')
57
+ embedding_path = hf_hub_download(repo_id='cwhuh/babyface_flux_dlora_3_Vietnamese', filename='/nas/checkpoints/sangmin/babyface_flux_dlora_3_Vietnamese_emb.safetensors', repo_type="model")
58
+ state_dict = load_file(embedding_path)
59
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>", "<s2>", "<s3>", "<s4>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
60
+
61
+ image = pipeline('A newborn <s0><s1><s2><s3><s4> baby.').images[0]
62
+ ```
63
+
64
+ 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)
65
+
66
+ ## License
67
+
68
+ Please adhere to the licensing terms as described [here](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md).
69
+
70
+
71
+ ## Intended uses & limitations
72
+
73
+ #### How to use
74
+
75
+ ```python
76
+ # TODO: add an example code snippet for running this diffusion pipeline
77
+ ```
78
+
79
+ #### Limitations and bias
80
+
81
+ [TODO: provide examples of latent issues and potential remediations]
82
+
83
+ ## Training details
84
+
85
+ [TODO: describe the data used to train the model]
babyface_flux_dlora_3_Vietnamese_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61d0e3d620fbb2979749d9d24a52d9d777bf1097e1cf83b6ac0d66d74c9eb7cf
3
+ size 15440
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df9987e4ba8cdb45e2842283dbaa2d5154c72f647f64dbeba426e8730096fff9
3
+ size 640311878
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:990a66cd6eaa7cc11f514814515222387439f945fca2660384e6916986fe5812
3
+ size 9388696
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58a0e87640b115a47d33a4e5d9c81a390096efd9baa02fca4ba6b7933f569feb
3
+ size 14344
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a288a1d370d465c283a1c2547626e53e4abcc6aa2675d8cc31e9f13960aa26f
3
+ size 1064
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5437c9da93946d2b0e961e16368b9caac51e0cc15749fddf25250883102b6c2b
3
+ size 640311878
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e9ed3cde3564fac717631a7c1e35098a32e044433782fc30ac9c067db066289
3
+ size 9388696
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c1a5e658bc71fcd99f639c8eecb9d3505857badee0f2838b6c1907620a05e22
3
+ size 14344
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56833320af79f6404a460f314f644e2513097895b5b0ae91f12f6a781b1f3280
3
+ size 1064
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:990a66cd6eaa7cc11f514814515222387439f945fca2660384e6916986fe5812
3
+ size 9388696