End of training
Browse files- README.md +87 -0
- babyface_flux_dlora_hsfw_African_American_emb.safetensors +3 -0
- checkpoint-1000/optimizer.bin +3 -0
- checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1000/random_states_0.pkl +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-1500/optimizer.bin +3 -0
- checkpoint-1500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1500/random_states_0.pkl +3 -0
- checkpoint-1500/scheduler.bin +3 -0
- checkpoint-2000/optimizer.bin +3 -0
- checkpoint-2000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2000/random_states_0.pkl +3 -0
- checkpoint-2000/scheduler.bin +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scheduler.bin +3 -0
- pytorch_lora_weights.safetensors +3 -0
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
|