Commit
·
89c0941
1
Parent(s):
89c12e7
Upload . with huggingface_hub
Browse files- README.md +22 -0
- checkpoint-100/custom_checkpoint_0.pkl +3 -0
- checkpoint-100/optimizer.bin +3 -0
- checkpoint-100/pytorch_model.bin +3 -0
- checkpoint-100/random_states_0.pkl +3 -0
- checkpoint-100/scaler.pt +3 -0
- checkpoint-100/scheduler.bin +3 -0
- logs/dreambooth-lora/1678220310.9551847/events.out.tfevents.1678220310.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.23442.1 +3 -0
- logs/dreambooth-lora/1678220310.9572747/hparams.yml +52 -0
- logs/dreambooth-lora/events.out.tfevents.1678220310.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.23442.0 +3 -0
- pytorch_lora_weights.bin +3 -0
- test_images/image_0.png +0 -0
- test_images/image_1.png +0 -0
- test_images/image_2.png +0 -0
- test_images/image_3.png +0 -0
- train.sh +1 -0
README.md
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: creativeml-openrail-m
|
3 |
+
base_model: andite/pastel-mix
|
4 |
+
instance_prompt: haku(naruto), haku, female, woman, girl, 1 girl, flat breasts, black eyes, brown eyes, mask, anbu mask, white and red mask, white mask, red mask, holding mask, headband, solid steel headband, symbol headband, white hair bun, hair bun, black hair, long hair, needles, holding needles, solid steel needles, steel needles, kimono, teal kimono, battlesuit, brown battlesuit, long sleeves, tassel
|
5 |
+
tags:
|
6 |
+
- stable-diffusion
|
7 |
+
- stable-diffusion-diffusers
|
8 |
+
- text-to-image
|
9 |
+
- diffusers
|
10 |
+
- lora
|
11 |
+
inference: true
|
12 |
+
---
|
13 |
+
# LoRA DreamBooth - haku-naruto
|
14 |
+
|
15 |
+
These are LoRA adaption weights for [andite/pastel-mix](https://huggingface.co/andite/pastel-mix). The weights were trained on the instance prompt "haku(naruto), haku, female, woman, girl, 1 girl, flat breasts, black eyes, brown eyes, mask, anbu mask, white and red mask, white mask, red mask, holding mask, headband, solid steel headband, symbol headband, white hair bun, hair bun, black hair, long hair, needles, holding needles, solid steel needles, steel needles, kimono, teal kimono, battlesuit, brown battlesuit, long sleeves, tassel" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
|
16 |
+
|
17 |
+
Test prompt: haku(naruto), haku, female, woman, girl, 1 girl, flat breasts
|
18 |
+

|
19 |
+

|
20 |
+

|
21 |
+

|
22 |
+
|
checkpoint-100/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:17419371aa50c9c5f67863153c02d798ed1d00c3e30cfe65c147dab7e63fe098
|
3 |
+
size 3287513
|
checkpoint-100/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:77802f17049c416190a6deb88d57b366b89b22224d002cb66b487ded9fde85c1
|
3 |
+
size 4000965
|
checkpoint-100/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:770742ad25eb1c26290943c636009039025cf8b2e03069f2bf19bc1506ff263a
|
3 |
+
size 3285965
|
checkpoint-100/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b20102cb107c592046ca08ae9260245869e1c39d254752014ad0649c86668028
|
3 |
+
size 14599
|
checkpoint-100/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cfa44e8523f62833816d29aa6c576eaa7783e3bbdb3e132e248b1d8aaee6132b
|
3 |
+
size 557
|
checkpoint-100/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:906bdd391189fd16d3160fd1e198eb53687dae580f930cf3f64365609fd5f606
|
3 |
+
size 563
|
logs/dreambooth-lora/1678220310.9551847/events.out.tfevents.1678220310.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.23442.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:da637d80dbe7e04dd7f36957252417038d3f132fc6974454596f76056945f71f
|
3 |
+
size 2852
|
logs/dreambooth-lora/1678220310.9572747/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
adam_beta1: 0.9
|
2 |
+
adam_beta2: 0.999
|
3 |
+
adam_epsilon: 1.0e-08
|
4 |
+
adam_weight_decay: 0.01
|
5 |
+
allow_tf32: false
|
6 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 100
|
8 |
+
class_data_dir: null
|
9 |
+
class_prompt: null
|
10 |
+
dataloader_num_workers: 0
|
11 |
+
enable_xformers_memory_efficient_attention: false
|
12 |
+
gradient_accumulation_steps: 1
|
13 |
+
gradient_checkpointing: false
|
14 |
+
hub_model_id: null
|
15 |
+
hub_token: null
|
16 |
+
instance_data_dir: /home/user/app/training_data/haku-naruto
|
17 |
+
instance_prompt: haku(naruto), haku, female, woman, girl, 1 girl, flat breasts, black
|
18 |
+
eyes, brown eyes, mask, anbu mask, white and red mask, white mask, red mask, holding
|
19 |
+
mask, headband, solid steel headband, symbol headband, white hair bun, hair bun,
|
20 |
+
black hair, long hair, needles, holding needles, solid steel needles, steel needles,
|
21 |
+
kimono, teal kimono, battlesuit, brown battlesuit, long sleeves, tassel
|
22 |
+
learning_rate: 0.0001
|
23 |
+
local_rank: -1
|
24 |
+
logging_dir: logs
|
25 |
+
lr_num_cycles: 1
|
26 |
+
lr_power: 1.0
|
27 |
+
lr_scheduler: constant
|
28 |
+
lr_warmup_steps: 0
|
29 |
+
max_grad_norm: 1.0
|
30 |
+
max_train_steps: 150
|
31 |
+
mixed_precision: fp16
|
32 |
+
num_class_images: 100
|
33 |
+
num_train_epochs: 12
|
34 |
+
num_validation_images: 4
|
35 |
+
output_dir: /home/user/app/experiments/haku-naruto
|
36 |
+
pretrained_model_name_or_path: andite/pastel-mix
|
37 |
+
prior_generation_precision: null
|
38 |
+
prior_loss_weight: 1.0
|
39 |
+
push_to_hub: false
|
40 |
+
report_to: tensorboard
|
41 |
+
resolution: 512
|
42 |
+
resume_from_checkpoint: null
|
43 |
+
revision: null
|
44 |
+
sample_batch_size: 4
|
45 |
+
scale_lr: false
|
46 |
+
seed: 0
|
47 |
+
tokenizer_name: null
|
48 |
+
train_batch_size: 1
|
49 |
+
use_8bit_adam: true
|
50 |
+
validation_epochs: 100
|
51 |
+
validation_prompt: haku(naruto), haku, female, woman, girl, 1 girl, flat breasts
|
52 |
+
with_prior_preservation: false
|
logs/dreambooth-lora/events.out.tfevents.1678220310.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.23442.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:600b1c91c90ed85cb4575cfe7351ad7649f9be4e0f4ca12fd71e61f6d1c1b15d
|
3 |
+
size 869219
|
pytorch_lora_weights.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cb6215f11594e867b801cb8ccf3fefd22825095f14a8849aa4510d69a305367a
|
3 |
+
size 3287771
|
test_images/image_0.png
ADDED
![]() |
test_images/image_1.png
ADDED
![]() |
test_images/image_2.png
ADDED
![]() |
test_images/image_3.png
ADDED
![]() |
train.sh
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
accelerate launch train_dreambooth_lora.py --pretrained_model_name_or_path=andite/pastel-mix --instance_data_dir=/home/user/app/training_data/haku-naruto --output_dir=/home/user/app/experiments/haku-naruto --instance_prompt="haku(naruto), haku, female, woman, girl, 1 girl, flat breasts, black eyes, brown eyes, mask, anbu mask, white and red mask, white mask, red mask, holding mask, headband, solid steel headband, symbol headband, white hair bun, hair bun, black hair, long hair, needles, holding needles, solid steel needles, steel needles, kimono, teal kimono, battlesuit, brown battlesuit, long sleeves, tassel" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=150 --checkpointing_steps=100 --validation_prompt="haku(naruto), haku, female, woman, girl, 1 girl, flat breasts" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam
|