End of training
Browse files- README.md +43 -0
- checkpoint-1000/model.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/model.safetensors +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/model.safetensors +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-2500/model.safetensors +3 -0
- checkpoint-2500/optimizer.bin +3 -0
- checkpoint-2500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2500/random_states_0.pkl +3 -0
- checkpoint-2500/scheduler.bin +3 -0
- checkpoint-3000/model.safetensors +3 -0
- checkpoint-3000/optimizer.bin +3 -0
- checkpoint-3000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-3000/random_states_0.pkl +3 -0
- checkpoint-3000/scheduler.bin +3 -0
- checkpoint-500/model.safetensors +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
- image_0.png +0 -0
- image_1.png +0 -0
- image_2.png +0 -0
- image_3.png +0 -0
- logs/text2image-fine-tune/1731851505.8166358/events.out.tfevents.1731851505.3dgs.1950715.1 +3 -0
- logs/text2image-fine-tune/1731851505.8175166/hparams.yml +50 -0
- logs/text2image-fine-tune/1731851574.1293952/events.out.tfevents.1731851574.3dgs.1951114.1 +3 -0
- logs/text2image-fine-tune/1731851574.1303124/hparams.yml +50 -0
- logs/text2image-fine-tune/events.out.tfevents.1731851505.3dgs.1950715.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1731851574.3dgs.1951114.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
ADDED
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: stable-diffusion-v1-5/stable-diffusion-v1-5
|
3 |
+
library_name: diffusers
|
4 |
+
license: creativeml-openrail-m
|
5 |
+
inference: true
|
6 |
+
tags:
|
7 |
+
- stable-diffusion
|
8 |
+
- stable-diffusion-diffusers
|
9 |
+
- text-to-image
|
10 |
+
- diffusers
|
11 |
+
- diffusers-training
|
12 |
+
- lora
|
13 |
+
---
|
14 |
+
|
15 |
+
<!-- This model card has been generated automatically according to the information the training script had access to. You
|
16 |
+
should probably proofread and complete it, then remove this comment. -->
|
17 |
+
|
18 |
+
|
19 |
+
# LoRA text2image fine-tuning - yijiu/output
|
20 |
+
These are LoRA adaption weights for stable-diffusion-v1-5/stable-diffusion-v1-5. The weights were fine-tuned on the yijiu/kanji dataset. You can find some example images in the following.
|
21 |
+
|
22 |
+
![img_0](./image_0.png)
|
23 |
+
![img_1](./image_1.png)
|
24 |
+
![img_2](./image_2.png)
|
25 |
+
![img_3](./image_3.png)
|
26 |
+
|
27 |
+
|
28 |
+
|
29 |
+
## Intended uses & limitations
|
30 |
+
|
31 |
+
#### How to use
|
32 |
+
|
33 |
+
```python
|
34 |
+
# TODO: add an example code snippet for running this diffusion pipeline
|
35 |
+
```
|
36 |
+
|
37 |
+
#### Limitations and bias
|
38 |
+
|
39 |
+
[TODO: provide examples of latent issues and potential remediations]
|
40 |
+
|
41 |
+
## Training details
|
42 |
+
|
43 |
+
[TODO: describe the data used to train the model]
|
checkpoint-1000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb3b1113e25fceac83058a19fd6321bff9a9a9532035da19ca4877f776ffab2a
|
3 |
+
size 3441397096
|
checkpoint-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f4516554fd3cc013d77c4248119ba1db494912727193f52383c31f79acf300e3
|
3 |
+
size 6592122
|
checkpoint-1000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6fcd9ed0d2064403e63c237d5608491c11d772828f2c4a16abe5910af9eb1d00
|
3 |
+
size 3226184
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:917e086910107fbf83e74413a1d0f261be9f7e1d75a371b4c9e2cf8054ec7f30
|
3 |
+
size 14408
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e1ef945fb67548939f1340070803ddc6f91925dd6d43f8c15ba25d113f03c5ed
|
3 |
+
size 1000
|
checkpoint-1500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:74e4dea07e3630b0c838c0bc2d77c12bbe5385d93af1b260e8760bc5fdec4621
|
3 |
+
size 3441397096
|
checkpoint-1500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:472f7bd1da1f2f9eb9db96e4f01b8523a71a59e4fd923524bef6b3170b8e22e7
|
3 |
+
size 6592122
|
checkpoint-1500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ff74b102e1e8df07c384c29389d3a091a24a2e822fe6f9260c668323c02108f
|
3 |
+
size 3226184
|
checkpoint-1500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fb9fdd0912862a8393fe61b53c8284dbbe5f1f8ffe46377a4c8db0954768dd7
|
3 |
+
size 14408
|
checkpoint-1500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:48bc4e7f165d13c3a118fbfce412755c802dfee9a7a91ae2a24ef2a694c18483
|
3 |
+
size 1000
|
checkpoint-2000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a1c319e1566e184d64cece3a2b3d00314d091d598aae94ba35d3557081a210b2
|
3 |
+
size 3441397096
|
checkpoint-2000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:693328b523d35c683bc0a859d3cd42b03afa0bac57c7e2529cc86a799d0daf44
|
3 |
+
size 6592122
|
checkpoint-2000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ee655dcacd12d132bcbc20cdc6f478c2a44c976e302656a9097d65aa67e5bd8
|
3 |
+
size 3226184
|
checkpoint-2000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f0a1de784157e7a0e839c9049e66e99e495444dfc89f0777c561d429b9766612
|
3 |
+
size 14408
|
checkpoint-2000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5a14cab18e3cf6bf4020ba19261b57bd47f02eb286b8412b0d80f0e1c055663a
|
3 |
+
size 1000
|
checkpoint-2500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4aeda7d978dac049998166a6ac9a8896399dbf680d7a781ecede1ccdf7401f46
|
3 |
+
size 3441397096
|
checkpoint-2500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:523a318cff71939c54146bc4c6770fc84d0a44af049ca4142989cd649eb2fad3
|
3 |
+
size 6592122
|
checkpoint-2500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c295665ca2e84dcc12767eadfefb3293c246cbb2a93bc3369e29a11d9edef1d9
|
3 |
+
size 3226184
|
checkpoint-2500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e422f8fda5780ad3ca65a02f30b79bf21e65ad7b38ca5989b5df0b8058971645
|
3 |
+
size 14408
|
checkpoint-2500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7806de3d060880702efffb6e8dfae4e935b646ac22a298bfd34612aa6bd7e66a
|
3 |
+
size 1000
|
checkpoint-3000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8cef442f48f4fb36948af64360f2ba17dce5d5b5de4d9a56ed3c927e2e639ad8
|
3 |
+
size 3441397096
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3e7d9ac152805fd469d9020be31ee71434a7fe4ba48041df3ef192c8aaa0d3f7
|
3 |
+
size 6592122
|
checkpoint-3000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cb029daa87b5fdc6b3040d0aa15bc6cb90e0faa41aaa7b56e0277fad2bf9b924
|
3 |
+
size 3226184
|
checkpoint-3000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0e87a1c43dd4b2f05492a81229cfa1238ba256f4072946c53af0c3e772e6cd83
|
3 |
+
size 14408
|
checkpoint-3000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2833302ff3e9f19a529d4b4c25d428a2569db9e059211a2009fd20f9a2843aba
|
3 |
+
size 1000
|
checkpoint-500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c2c9d4467d0ebf69c9bd38b55d1f55b527c795ca535b49f2ff0178054f6dddc2
|
3 |
+
size 3441397096
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ab8568230623be22cdd2186adbe350e2b80e5234843694c7d37ff6ae0e5921b4
|
3 |
+
size 6592122
|
checkpoint-500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6a0169167bbe9179c19ae16a4447fc1b4275699fa6c439f60ab4eff836849082
|
3 |
+
size 3226184
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4405c33538006d0d660f6c8875e5fd86cf593590404bfcabee814f478bb10bfa
|
3 |
+
size 14408
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c
|
3 |
+
size 1000
|
image_0.png
ADDED
image_1.png
ADDED
image_2.png
ADDED
image_3.png
ADDED
logs/text2image-fine-tune/1731851505.8166358/events.out.tfevents.1731851505.3dgs.1950715.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:039334de4a8d09bbcdf5de87f796ca8199f0216ecb90f744f01e6d2acbf3f3bc
|
3 |
+
size 2206
|
logs/text2image-fine-tune/1731851505.8175166/hparams.yml
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
cache_dir: null
|
7 |
+
caption_column: text
|
8 |
+
center_crop: false
|
9 |
+
checkpointing_steps: 500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 0
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: yijiu/kanji
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: null
|
18 |
+
hub_token: null
|
19 |
+
image_column: image
|
20 |
+
learning_rate: 0.0001
|
21 |
+
local_rank: -1
|
22 |
+
logging_dir: logs
|
23 |
+
lr_scheduler: constant
|
24 |
+
lr_warmup_steps: 0
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 3206
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 2
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: output
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
|
35 |
+
push_to_hub: true
|
36 |
+
random_flip: true
|
37 |
+
rank: 4
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 512
|
40 |
+
resume_from_checkpoint: null
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 42
|
44 |
+
snr_gamma: null
|
45 |
+
train_batch_size: 4
|
46 |
+
train_data_dir: null
|
47 |
+
use_8bit_adam: false
|
48 |
+
validation_epochs: 1
|
49 |
+
validation_prompt: forest
|
50 |
+
variant: null
|
logs/text2image-fine-tune/1731851574.1293952/events.out.tfevents.1731851574.3dgs.1951114.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b162d7ed72b1fe5d6e0185c6234949f0eb0464d0d1d5fdd41fa65400ff7e1ac6
|
3 |
+
size 2206
|
logs/text2image-fine-tune/1731851574.1303124/hparams.yml
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
cache_dir: null
|
7 |
+
caption_column: text
|
8 |
+
center_crop: false
|
9 |
+
checkpointing_steps: 500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 0
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: yijiu/kanji
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: null
|
18 |
+
hub_token: null
|
19 |
+
image_column: image
|
20 |
+
learning_rate: 0.0001
|
21 |
+
local_rank: -1
|
22 |
+
logging_dir: logs
|
23 |
+
lr_scheduler: constant
|
24 |
+
lr_warmup_steps: 0
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 3206
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 2
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: output
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
|
35 |
+
push_to_hub: true
|
36 |
+
random_flip: true
|
37 |
+
rank: 4
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 512
|
40 |
+
resume_from_checkpoint: null
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 42
|
44 |
+
snr_gamma: null
|
45 |
+
train_batch_size: 4
|
46 |
+
train_data_dir: null
|
47 |
+
use_8bit_adam: false
|
48 |
+
validation_epochs: 1
|
49 |
+
validation_prompt: forest
|
50 |
+
variant: null
|
logs/text2image-fine-tune/events.out.tfevents.1731851505.3dgs.1950715.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebf75c18b2be36e38af2793282fb2c72cbf563a3a414e8dc198fe6d8871a264b
|
3 |
+
size 88
|
logs/text2image-fine-tune/events.out.tfevents.1731851574.3dgs.1951114.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f7360faa3264b6fb4094b9914822643966ddf78061be83b4783b2c606b87a744
|
3 |
+
size 3820073
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:73bad4abe063ce999558694bbae7e6ba5a29c77a96ac335b616d357686d8d349
|
3 |
+
size 3226184
|