cwhuh commited on
Commit
4acb481
·
verified ·
1 Parent(s): 9ed0486

End of training

Browse files
README.md ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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><s5><s6><s7><s8><s9><s10><s11><s12><s13><s14><s15><s16><s17><s18><s19><s20><s21><s22><s23><s24><s25><s26><s27><s28><s29><s30><s31><s32><s33><s34><s35><s36><s37>
6
+ baby. The baby is wearing a white beanie and is swaddled in a white blanket. The
7
+ background is a soft, neutral white, matching the original clean studio aesthetic.
8
+ widget: []
9
+ tags:
10
+ - text-to-image
11
+ - diffusers-training
12
+ - diffusers
13
+ - lora
14
+ - flux
15
+ - flux-diffusers
16
+ - template:sd-lora
17
+ ---
18
+
19
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
20
+ should probably proofread and complete it, then remove this comment. -->
21
+
22
+
23
+ # Flux DreamBooth LoRA - cwhuh/babyface_flux_dlora_hsfw_Hispanic
24
+
25
+ <Gallery />
26
+
27
+ ## Model description
28
+
29
+ These are cwhuh/babyface_flux_dlora_hsfw_Hispanic DreamBooth LoRA weights for black-forest-labs/FLUX.1-dev.
30
+
31
+ 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).
32
+
33
+ Was LoRA for the text encoder enabled? False.
34
+
35
+ Pivotal tuning was enabled: True.
36
+
37
+ ## Trigger words
38
+
39
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
40
+
41
+ to trigger concept `hsfw` → use `<s0><s1><s2><s3><s4><s5><s6><s7><s8><s9><s10><s11><s12><s13><s14><s15><s16><s17><s18><s19><s20><s21><s22><s23><s24><s25><s26><s27><s28><s29><s30><s31><s32><s33><s34><s35><s36><s37>` in your prompt
42
+
43
+
44
+
45
+ ## Download model
46
+
47
+ [Download the *.safetensors LoRA](cwhuh/babyface_flux_dlora_hsfw_Hispanic/tree/main) in the Files & versions tab.
48
+
49
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
50
+
51
+ ```py
52
+ from diffusers import AutoPipelineForText2Image
53
+ import torch
54
+ from huggingface_hub import hf_hub_download
55
+ from safetensors.torch import load_file
56
+
57
+ pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
58
+ pipeline.load_lora_weights('cwhuh/babyface_flux_dlora_hsfw_Hispanic', weight_name='pytorch_lora_weights.safetensors')
59
+ embedding_path = hf_hub_download(repo_id='cwhuh/babyface_flux_dlora_hsfw_Hispanic', filename='/nas/checkpoints/sangmin/babyface_flux_dlora_hsfw_Hispanic_emb.safetensors', repo_type="model")
60
+ state_dict = load_file(embedding_path)
61
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>", "<s2>", "<s3>", "<s4>", "<s5>", "<s6>", "<s7>", "<s8>", "<s9>", "<s10>", "<s11>", "<s12>", "<s13>", "<s14>", "<s15>", "<s16>", "<s17>", "<s18>", "<s19>", "<s20>", "<s21>", "<s22>", "<s23>", "<s24>", "<s25>", "<s26>", "<s27>", "<s28>", "<s29>", "<s30>", "<s31>", "<s32>", "<s33>", "<s34>", "<s35>", "<s36>", "<s37>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
62
+
63
+ image = pipeline('A newborn <s0><s1><s2><s3><s4><s5><s6><s7><s8><s9><s10><s11><s12><s13><s14><s15><s16><s17><s18><s19><s20><s21><s22><s23><s24><s25><s26><s27><s28><s29><s30><s31><s32><s33><s34><s35><s36><s37> baby. The baby is wearing a white beanie and is swaddled in a white blanket. The background is a soft, neutral white, matching the original clean studio aesthetic.').images[0]
64
+ ```
65
+
66
+ 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)
67
+
68
+ ## License
69
+
70
+ Please adhere to the licensing terms as described [here](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md).
71
+
72
+
73
+ ## Intended uses & limitations
74
+
75
+ #### How to use
76
+
77
+ ```python
78
+ # TODO: add an example code snippet for running this diffusion pipeline
79
+ ```
80
+
81
+ #### Limitations and bias
82
+
83
+ [TODO: provide examples of latent issues and potential remediations]
84
+
85
+ ## Training details
86
+
87
+ [TODO: describe the data used to train the model]
babyface_flux_dlora_hsfw_Hispanic_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0b7c8a99c4d5ddc13d9021b74417b0c78fdecc4da0fc6c879f3e758b18c7fe8
3
+ size 116816
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97a43b005849db09ed4970c0b3474e0ee91fbbda889dea0fb521e634f2c3c08a
3
+ size 640717382
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2106b34006724cf0815b9f417374cf4bf45b91dcd842bd5c7efaed586198b394
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:7b127af1150a800390a78fc4da4f1e14ed78edcfd067c2d971982226079cfbb4
3
+ size 14344
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4422049d48e28b0876bfdb9b3c22b8f899bf2a57097aae313198898276971bd
3
+ size 1064
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5de789ee915e3a563dc1fcc3aba4540ec6eb79a76e644f2c0cf815e2aa7b05a9
3
+ size 640717382
checkpoint-1500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd79c80d95b655334faf587e75f090eb5ef4233cc27b68d2c8d49a2b0cf284ec
3
+ size 9388696
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:262b27cae1ea3d1efeabd89a48c2df698a2451c7fe6e3cfeb37fbb32cd181e43
3
+ size 14344
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:168304fdafc208b0dcca5471ef496eb7604d7adf3623e21855dae97350aa9f01
3
+ size 1064
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f1431caf6caf57c49a7fba874125ad189c9f7db4a3513177b95fa481f460eb9
3
+ size 640717382
checkpoint-2000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cb6619f898c52425b233345ff5dd163af6c624f36dd38ac0578cdbfb18ba3b3
3
+ size 9388696
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e494330ff451ac34081be84ace36ff8a5f59bbd91ec0b5a61f4a4e67f15160f4
3
+ size 14344
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae1adce372e909497d857cb683296a9693c102e08c923a9847f4762ae4ce4b7b
3
+ size 1064
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec9e219b9e98f73fb202789492de1314f9d7bd76e0cffb756633109ed1d7dce1
3
+ size 640717382
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4e86f1e771b4c9349e84f50a5624ab34affb4492ddff7ddb7175faa59e5f623
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:ce062eb28572a51ca025502613550f2bc1817564be42870cc3fc56f16dcb382e
3
+ size 14344
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6b8e2b5f93eeee327604455b96d2f2e8563803876d26c00a38a3f457d35bc1b
3
+ size 1064
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cb6619f898c52425b233345ff5dd163af6c624f36dd38ac0578cdbfb18ba3b3
3
+ size 9388696