End of training
Browse files- README.md +23 -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
- checkpoint-200/custom_checkpoint_0.pkl +3 -0
- checkpoint-200/optimizer.bin +3 -0
- checkpoint-200/pytorch_model.bin +3 -0
- checkpoint-200/random_states_0.pkl +3 -0
- checkpoint-200/scaler.pt +3 -0
- checkpoint-200/scheduler.bin +3 -0
- checkpoint-300/custom_checkpoint_0.pkl +3 -0
- checkpoint-300/optimizer.bin +3 -0
- checkpoint-300/pytorch_model.bin +3 -0
- checkpoint-300/random_states_0.pkl +3 -0
- checkpoint-300/scaler.pt +3 -0
- checkpoint-300/scheduler.bin +3 -0
- checkpoint-400/custom_checkpoint_0.pkl +3 -0
- checkpoint-400/optimizer.bin +3 -0
- checkpoint-400/pytorch_model.bin +3 -0
- checkpoint-400/random_states_0.pkl +3 -0
- checkpoint-400/scaler.pt +3 -0
- checkpoint-400/scheduler.bin +3 -0
- checkpoint-500/custom_checkpoint_0.pkl +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_model.bin +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scaler.pt +3 -0
- checkpoint-500/scheduler.bin +3 -0
- image_0.png +0 -0
- image_1.png +0 -0
- image_2.png +0 -0
- image_3.png +0 -0
- logs/dreambooth-lora/1681837769.4736667/events.out.tfevents.1681837769.stable-diffusion-4-a100.652320.1 +3 -0
- logs/dreambooth-lora/1681837769.4753191/hparams.yml +49 -0
- logs/dreambooth-lora/events.out.tfevents.1681837769.stable-diffusion-4-a100.652320.0 +3 -0
- pytorch_lora_weights.bin +3 -0
README.md
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
license: creativeml-openrail-m
|
4 |
+
base_model: runwayml/stable-diffusion-v1-5
|
5 |
+
instance_prompt: a photo of sofa
|
6 |
+
tags:
|
7 |
+
- stable-diffusion
|
8 |
+
- stable-diffusion-diffusers
|
9 |
+
- text-to-image
|
10 |
+
- diffusers
|
11 |
+
- lora
|
12 |
+
inference: true
|
13 |
+
---
|
14 |
+
|
15 |
+
# LoRA DreamBooth - willhx/train_lora
|
16 |
+
|
17 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were trained on a photo of sofa using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
|
18 |
+
|
19 |
+
![img_0](./image_0.png)
|
20 |
+
![img_1](./image_1.png)
|
21 |
+
![img_2](./image_2.png)
|
22 |
+
![img_3](./image_3.png)
|
23 |
+
|
checkpoint-100/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3f37909f6a687470cc25a037b9d896aa2d5b333c6dd935be7a0e045e4b4d0c99
|
3 |
+
size 3287513
|
checkpoint-100/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b477d0df66e5ce11fa1409806645fa4d1966c8c47cb43eb1e3ea41bdfa009dd5
|
3 |
+
size 6591685
|
checkpoint-100/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:346816c85c7fb92a5bfee999db80c040c14f3ab33d74fe770c0211d6e0294d06
|
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:128e51c863a14b2901c024cffe02a37f887914e565a902269be77f4911ab6f1b
|
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
|
checkpoint-200/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ff53fda585ad484a06a32917c800e12e37c0e1bc7ba958136f0ef5163162200c
|
3 |
+
size 3287513
|
checkpoint-200/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fc2882f3ffdf9952d194ae0fc8e2e1ded95eda71643e20d032c3f170cb33df97
|
3 |
+
size 6591685
|
checkpoint-200/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b2cece936101d6806a20fc9403b2abd789c99967fbbbb8cc231d49c5a380bb61
|
3 |
+
size 3285965
|
checkpoint-200/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:93d62c49a43ef8ed24a2529887987fec7c474e7012a5574de0dcce9e878c5dfb
|
3 |
+
size 14599
|
checkpoint-200/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fccf0f9be1bb8f24861e4393745b3e09cc2687125a69e3757955fb0f0925ea5
|
3 |
+
size 557
|
checkpoint-200/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2565f1f317aadcb95557955732a7a05efb023cacf90620d89fcb0c7ac0f8e04
|
3 |
+
size 563
|
checkpoint-300/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bbc790f98ec9a018b5d9f94cd33d127c5f02c2fd1eb1bcee497b90cb7297b429
|
3 |
+
size 3287513
|
checkpoint-300/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b6e7af47c4e293ceefdd33e5b7241f8055ca2ec051ae5e94dc5c8cec214eec41
|
3 |
+
size 6591685
|
checkpoint-300/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67934c0b9b8cd143b93431dbab242659332da74d0285747b026581688fecdc29
|
3 |
+
size 3285965
|
checkpoint-300/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e536b58526fe981d6fa218fd7edd4b9cc547759aafd77417f175b6edd5cabb89
|
3 |
+
size 14599
|
checkpoint-300/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:efafd90182e3d39d1b7c4a686f86e5913f5abc094dc3e2f827a6d479c6cef247
|
3 |
+
size 557
|
checkpoint-300/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a84fb1fc02dc9adb8c44beeab2c46fbc41e10a07e99dad3fcd85350ca2bda9d5
|
3 |
+
size 563
|
checkpoint-400/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1222dfbd4c7f7833718cdb3c80da065fe7fdfb8de7302c8ad9e636337639396f
|
3 |
+
size 3287513
|
checkpoint-400/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7bb0ba3989cac9910132e9ba43371ff90669f782412f6046d9c4f29362327d0c
|
3 |
+
size 6591685
|
checkpoint-400/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ee1eb37b0fcb33c91dc0e70b4185314c50b6f1325f0fbefc3376a17caf26145d
|
3 |
+
size 3285965
|
checkpoint-400/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5af9c8d4ae86ea9605325bbc6d16d42292bbbefb40c49f12323227c47a916683
|
3 |
+
size 14599
|
checkpoint-400/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cc40a4be6a52cee4d7658df4041e660ffa02b0d8b5bd143bb8bb397f7b71b1a5
|
3 |
+
size 557
|
checkpoint-400/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5e33b45756ec30d1126f6571b94c7435f6d9cb7b6d5e5e650c81523775e7216f
|
3 |
+
size 563
|
checkpoint-500/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f867765094493693e9c6ee68a14b800ac15dd33e49ae9481e015807d5f1b3727
|
3 |
+
size 3287513
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19ffda2cf601a78abdb825bc0f0dd02796a078f50443fffc59ad517529ba3e1e
|
3 |
+
size 6591685
|
checkpoint-500/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:142675a16fb8b242683e69da80fc1f9a65f5b5ddae1cf68a9b8c8570f316dcfc
|
3 |
+
size 3285965
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1e70ab7442d77ab99c4a62864562d29cf6105a6f9035ca249b737c0fd98f8a77
|
3 |
+
size 14599
|
checkpoint-500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
|
3 |
+
size 557
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:62783ff6f0ae2c2ccc177b097e430c03940c5e70c38a191df273c5cf7ab1227c
|
3 |
+
size 563
|
image_0.png
ADDED
image_1.png
ADDED
image_2.png
ADDED
image_3.png
ADDED
logs/dreambooth-lora/1681837769.4736667/events.out.tfevents.1681837769.stable-diffusion-4-a100.652320.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:60e847b09d7c57f41b1526b2578f6b943364ad196c7af52b94bf6deda75d24fe
|
3 |
+
size 2433
|
logs/dreambooth-lora/1681837769.4753191/hparams.yml
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
dataloader_num_workers: 0
|
12 |
+
enable_xformers_memory_efficient_attention: false
|
13 |
+
gradient_accumulation_steps: 1
|
14 |
+
gradient_checkpointing: false
|
15 |
+
hub_model_id: null
|
16 |
+
hub_token: null
|
17 |
+
instance_data_dir: /root/whx/sofa
|
18 |
+
instance_prompt: a photo of sofa
|
19 |
+
learning_rate: 0.0001
|
20 |
+
local_rank: -1
|
21 |
+
logging_dir: logs
|
22 |
+
lr_num_cycles: 1
|
23 |
+
lr_power: 1.0
|
24 |
+
lr_scheduler: constant
|
25 |
+
lr_warmup_steps: 0
|
26 |
+
max_grad_norm: 1.0
|
27 |
+
max_train_steps: 500
|
28 |
+
mixed_precision: null
|
29 |
+
num_class_images: 100
|
30 |
+
num_train_epochs: 100
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: /root/whx/diffusers/examples/dreambooth/output/train_lora
|
33 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
34 |
+
prior_generation_precision: null
|
35 |
+
prior_loss_weight: 1.0
|
36 |
+
push_to_hub: true
|
37 |
+
report_to: tensorboard
|
38 |
+
resolution: 512
|
39 |
+
resume_from_checkpoint: null
|
40 |
+
revision: null
|
41 |
+
sample_batch_size: 4
|
42 |
+
scale_lr: false
|
43 |
+
seed: 0
|
44 |
+
tokenizer_name: null
|
45 |
+
train_batch_size: 1
|
46 |
+
use_8bit_adam: false
|
47 |
+
validation_epochs: 50
|
48 |
+
validation_prompt: A photo of sofa
|
49 |
+
with_prior_preservation: false
|
logs/dreambooth-lora/events.out.tfevents.1681837769.stable-diffusion-4-a100.652320.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3da1ae3e3ce345c8731e30ecd9c26255844131e273de372dbdee5ab96a36f615
|
3 |
+
size 4834534
|
pytorch_lora_weights.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4812c0faacfb842cce1b62127b0b570fcbedb78568ceaf9d0ee927516bad6b50
|
3 |
+
size 3287771
|