antonellaavad commited on
Commit
89c0941
1 Parent(s): 89c12e7

Upload . with huggingface_hub

Browse files
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
+ ![image_0](test_images/image_0.png)
19
+ ![image_1](test_images/image_1.png)
20
+ ![image_2](test_images/image_2.png)
21
+ ![image_3](test_images/image_3.png)
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