cwhuh commited on
Commit
d169ab2
·
verified ·
1 Parent(s): 0fb3af8

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><s38>
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_African_American
24
+
25
+ <Gallery />
26
+
27
+ ## Model description
28
+
29
+ These are cwhuh/babyface_flux_dlora_hsfw_African_American 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><s38>` in your prompt
42
+
43
+
44
+
45
+ ## Download model
46
+
47
+ [Download the *.safetensors LoRA](cwhuh/babyface_flux_dlora_hsfw_African_American/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_African_American', weight_name='pytorch_lora_weights.safetensors')
59
+ embedding_path = hf_hub_download(repo_id='cwhuh/babyface_flux_dlora_hsfw_African_American', filename='/nas/checkpoints/sangmin/babyface_flux_dlora_hsfw_African_American_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>", "<s38>"], 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><s38> 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_African_American_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed6c230efe59c7c4cc6b21ac11cd72db50be34809ce283c30122186b2827f275
3
+ size 119888
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19ae90ec5de04c91a04c73d4664b3a3fc7476ca2af4d4b2cd9d987b87b70ce83
3
+ size 640729670
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b292d6fb919a37d591b3b7b9cb6cdf4e1e1352f5ad0d17a7b33c6568c8db1de7
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:f667c1490773e06f3670a3292fdd7451f878af7165bc6a1beeda9f1a17746c51
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:cf03d44050be0ca90defd4d554a8eef9b28cbce0c8aceae2d73a32edcb3b3675
3
+ size 640729670
checkpoint-1500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d94b5b2a43b2b12469d8a6732c48c1cc63fb357f69a478674603d75fe53e1595
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:4d6816cd528d8cf9b1fbd43b8aafedc376b5dc7f934f0838ee29284ce79e7fbb
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:a760c5f9cbd4082cd4d7a4e3752a2ef33d723ac5c836dac6fcc14b1d4e3eb4dc
3
+ size 640729670
checkpoint-2000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aeaf8808319030f11e5b569a0860208fb86944d22a37e90d5dce41e033c44c77
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:ae10064a8e7ef844ee3ea57b0124b84665e34076e14251d2ebda59ddc509b04c
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:6a31e961f4aff8698f3ddb9d6ca9a560c91bf04523cfde3df87d34ebeec95cb3
3
+ size 640729670
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:995212d5574709837908d646d8a7f5803224cca412a7a76075e91b61246c6426
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:62c480f5f3b0be34c8dd63c7768ff060264f611dd3254926631e4ad7c409b42b
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:aeaf8808319030f11e5b569a0860208fb86944d22a37e90d5dce41e033c44c77
3
+ size 9388696