End of training
Browse files- README.md +85 -0
- babyface_flux_dlora_hsfw_White_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,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_hsfw_White
|
22 |
+
|
23 |
+
<Gallery />
|
24 |
+
|
25 |
+
## Model description
|
26 |
+
|
27 |
+
These are cwhuh/babyface_flux_dlora_hsfw_White 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 `White_hsfw` → 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_hsfw_White/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_hsfw_White', weight_name='pytorch_lora_weights.safetensors')
|
57 |
+
embedding_path = hf_hub_download(repo_id='cwhuh/babyface_flux_dlora_hsfw_White', filename='/nas/checkpoints/sangmin/babyface_flux_dlora_hsfw_White_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_hsfw_White_emb.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f634e979998bc6eb1615b81f50d16378386bda9b3fb418f1c3bd59b1fad384f7
|
3 |
+
size 15440
|
checkpoint-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9a9ebc267dd7ea061319c7ce617cd315ef9fc4a9c931153bc94cc6d50af30779
|
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:8ea5592ebc94fff976d5a6e6e5fccff8952f66e14a63f532d4885d89238b5d93
|
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:0d677b5275fc7676af55e902f0c52f792cdea150c0202f1e55068dea3b06ba9d
|
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:75c4c8c1b1f2736e9c7450c84db5f88c8eab030adc65f5a8027d4daa9e2bf231
|
3 |
+
size 640311878
|
checkpoint-1500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2a65df1714cf5ba5022adccfce8d6628e34e9d5129d694707e1111c2a2ec95c6
|
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:f3c223409d156ed459c8554d85d47b8adaa5a3fa436fdf1f8387cb32186b8bf7
|
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:77e14b6b400f6767743bab335481aa2f87fd61f1bed5238e5c97af5ed260b26b
|
3 |
+
size 640311878
|
checkpoint-2000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e6570dedd1324c1ff7a18deac5e67acf59f2c8d316960f1567d50f04aeac3e89
|
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:f6645707fee2015d93c80aabe4a9ee1e3d970eea4eb52e9c125441d6b7eee8d3
|
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:c42e7592a73ffe41e7a58914851b0abada4b910e3d46687bd0d41f49ccbee6b9
|
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:e020fe132c1cc29b233fd1840236b9955e83e602af0696f45c7478bff9ecfb9b
|
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:dad8fce9389b1d235e18958a4833b7652ebed8c925b346eb4ea86a7e44285e8d
|
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:e6570dedd1324c1ff7a18deac5e67acf59f2c8d316960f1567d50f04aeac3e89
|
3 |
+
size 9388696
|