End of training
Browse files- README.md +6 -0
- checkpoint-12000/model.safetensors +3 -0
- checkpoint-12000/optimizer.bin +3 -0
- checkpoint-12000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-12000/random_states_0.pkl +3 -0
- checkpoint-12000/scaler.pt +3 -0
- checkpoint-12000/scheduler.bin +3 -0
- checkpoint-15000/model.safetensors +3 -0
- checkpoint-15000/optimizer.bin +3 -0
- checkpoint-15000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-15000/random_states_0.pkl +3 -0
- checkpoint-15000/scaler.pt +3 -0
- checkpoint-15000/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/scaler.pt +3 -0
- checkpoint-3000/scheduler.bin +3 -0
- checkpoint-6000/model.safetensors +3 -0
- checkpoint-6000/optimizer.bin +3 -0
- checkpoint-6000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-6000/random_states_0.pkl +3 -0
- checkpoint-6000/scaler.pt +3 -0
- checkpoint-6000/scheduler.bin +3 -0
- checkpoint-9000/model.safetensors +3 -0
- checkpoint-9000/optimizer.bin +3 -0
- checkpoint-9000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-9000/random_states_0.pkl +3 -0
- checkpoint-9000/scaler.pt +3 -0
- checkpoint-9000/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/1720813367.3979056/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.1 +3 -0
- logs/text2image-fine-tune/1720813367.399584/hparams.yml +50 -0
- logs/text2image-fine-tune/1720834818.385705/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.1 +3 -0
- logs/text2image-fine-tune/1720834818.387928/hparams.yml +50 -0
- logs/text2image-fine-tune/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.0 +3 -0
- pytorch_lora_weights.safetensors +1 -1
README.md
CHANGED
@@ -9,6 +9,12 @@ tags:
|
|
9 |
- diffusers
|
10 |
- diffusers-training
|
11 |
- lora
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
inference: true
|
13 |
---
|
14 |
|
|
|
9 |
- diffusers
|
10 |
- diffusers-training
|
11 |
- lora
|
12 |
+
- stable-diffusion
|
13 |
+
- stable-diffusion-diffusers
|
14 |
+
- text-to-image
|
15 |
+
- diffusers
|
16 |
+
- diffusers-training
|
17 |
+
- lora
|
18 |
inference: true
|
19 |
---
|
20 |
|
checkpoint-12000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1f8fb2e045ecbb00b7ac6e79dd10643ed98b644fe0f5abd61ed4909a64c0a203
|
3 |
+
size 1722353544
|
checkpoint-12000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:25c2ded06af07056dbfb529f3cb9d18fdfaa701e9d0c39810a5e750f98cf3692
|
3 |
+
size 6592122
|
checkpoint-12000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:253d906c3199e0022103246f9fb5c119fa8b8a51b0015f495b8c35ad2fc1a15b
|
3 |
+
size 3226184
|
checkpoint-12000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a65a93f37b9bf6197d180eb449d5cbbfe678820d6c038e86e1bad931d8ab1ee9
|
3 |
+
size 14408
|
checkpoint-12000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2d1f9d6a63312bf6188d8844a7d621f2ecda20b6c0aee9e19ddcea1c480954ba
|
3 |
+
size 988
|
checkpoint-12000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9921b7da5c99c86f03a3f19361ffe4ae9a1d7472d81b1adc17eaae9fcbef933b
|
3 |
+
size 1000
|
checkpoint-15000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d57f7fec10a0efa182da5fe3269647bca984faeaa2b88683ce889793f934c843
|
3 |
+
size 1722353544
|
checkpoint-15000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9080f48edfd4f9a1068eb02f3c2148e08dce5f29307e87a2da1a4f9a729f6a18
|
3 |
+
size 6592122
|
checkpoint-15000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2780d5e3b74c027ab40240c1ee54adc178cb3f82929636a098e0db4e20f26f3b
|
3 |
+
size 3226184
|
checkpoint-15000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6d483c2bfb80aa778afd8aac3efabc5af3e0f72f9f667f8078925883b4a2ee4b
|
3 |
+
size 14408
|
checkpoint-15000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:80f2829dbe9102c995677cfbb16e04c7f863ba8bbe9c0f80b5f219d485af0321
|
3 |
+
size 988
|
checkpoint-15000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:25ac02ce31bea574afc7001488812dcd64ad3bff8601eb47775c323133594325
|
3 |
+
size 1000
|
checkpoint-3000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a12cd2ca8bfc6b097dfa3b05ce2ffe02266061ea1a8cc60c94576ac1b7914f25
|
3 |
+
size 1722353544
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2d61d8bcfe62ea188ab24dc67aab56a1d6de3c1a69dc5fc51fe3d09dbbc1872d
|
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:e4b25c6be6862e5a8ee3d70665352a971668e3d3d464e6e79baa2247e358a797
|
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:ddeaab2465529c41de91e04719cc693587c453d105f4056932d547087ed36432
|
3 |
+
size 14408
|
checkpoint-3000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:21aba8ed0f38ed1c04994c10a9ca7e9925e55ef2ed51283c43ff8e2cce78585f
|
3 |
+
size 988
|
checkpoint-3000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:89df8bcca59003290e3a854d22c2892d49d485440014cbb479fc1d4c852903d2
|
3 |
+
size 1000
|
checkpoint-6000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ce463b97700e2d92b996038e6457ef55e474b65cb27e516370c5ee91158bcd6d
|
3 |
+
size 1722353544
|
checkpoint-6000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3f97717f4f5ae0f4229357164ef674454e3bd77d8c62e55b1f0b62e197c48eed
|
3 |
+
size 6592122
|
checkpoint-6000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d0621810a41812125f73d4909fe5b7680b4ad5a81397a53c6408be8ed86fcc6
|
3 |
+
size 3226184
|
checkpoint-6000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fc9de4c60e5996f0ac0269285f6661eb9f2a316e495f3559829e24e7d532424c
|
3 |
+
size 14408
|
checkpoint-6000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23c6abec96b9d238cb94e3e6ca47ebac7366e65b2c1f7f780949153075612b6e
|
3 |
+
size 988
|
checkpoint-6000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a8efbe3d12cb8a569b24bc75287fb9269526a43518a997085b3fd06b850026b3
|
3 |
+
size 1000
|
checkpoint-9000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c5e5742c22e3f66aa24f185a01923928ee22f5745d331fb7bea07f64cebe9323
|
3 |
+
size 1722353544
|
checkpoint-9000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2d62607d1e13bfd7cbe199917375c4daab52ee3b022ddbd8ce377ce821c1f422
|
3 |
+
size 6592122
|
checkpoint-9000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:983d2fb68429c85adcdb097d188c59d3a92a0e86f8a8ee59cf5be5fc259042c0
|
3 |
+
size 3226184
|
checkpoint-9000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dda0f9b951447daef87560116be5b89f5a9c844e9dcd2c49a9cfb6b8261abb3e
|
3 |
+
size 14408
|
checkpoint-9000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ad6c50c35b53713927f28732052bf0f3bf5da247a85776559de9817d2d278c7
|
3 |
+
size 988
|
checkpoint-9000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b79c3422e1af76ea696e5b4dc00cd35dc6ee89af188fa2a23482f281e12789d
|
3 |
+
size 1000
|
image_0.png
CHANGED
image_1.png
CHANGED
image_2.png
CHANGED
image_3.png
CHANGED
logs/text2image-fine-tune/1720813367.3979056/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e76f06865c30b01cd00d896d91e22254df56a09abf1060ae1761bdc9b38aea58
|
3 |
+
size 2357
|
logs/text2image-fine-tune/1720813367.399584/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: true
|
9 |
+
checkpointing_steps: 3000
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 8
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: lambdalabs/naruto-blip-captions
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: true
|
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: cosine
|
24 |
+
lr_warmup_steps: 0
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 15000
|
28 |
+
mixed_precision: fp16
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 50
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: sd-naruto-model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/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: latest
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 1337
|
44 |
+
snr_gamma: null
|
45 |
+
train_batch_size: 1
|
46 |
+
train_data_dir: null
|
47 |
+
use_8bit_adam: false
|
48 |
+
validation_epochs: 1
|
49 |
+
validation_prompt: Stephen Curry with naruto style
|
50 |
+
variant: null
|
logs/text2image-fine-tune/1720834818.385705/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9f0dc22191fef9e949e3e7c82927e7c46e19100eb21cccde5a76d7d0fec8719f
|
3 |
+
size 2357
|
logs/text2image-fine-tune/1720834818.387928/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: true
|
9 |
+
checkpointing_steps: 3000
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 8
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: lambdalabs/naruto-blip-captions
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: true
|
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: cosine
|
24 |
+
lr_warmup_steps: 0
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 15000
|
28 |
+
mixed_precision: fp16
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 50
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: sd-naruto-model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/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: latest
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 1337
|
44 |
+
snr_gamma: null
|
45 |
+
train_batch_size: 1
|
46 |
+
train_data_dir: null
|
47 |
+
use_8bit_adam: false
|
48 |
+
validation_epochs: 1
|
49 |
+
validation_prompt: Stephen Curry with Naruto style
|
50 |
+
variant: null
|
logs/text2image-fine-tune/events.out.tfevents.1720813367.autodl-container-05c845b795-ce1e9284.17672.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dfb1c7835dcabc1424d5e3949029ee28d636714acb5053932f397e044094cc62
|
3 |
+
size 72467702
|
logs/text2image-fine-tune/events.out.tfevents.1720834818.autodl-container-05c845b795-ce1e9284.45823.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:26cd859ab7a1719c071635bc00533a9f890a80e83298d76c33c6edb5aba02220
|
3 |
+
size 20398868
|
pytorch_lora_weights.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3226184
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2780d5e3b74c027ab40240c1ee54adc178cb3f82929636a098e0db4e20f26f3b
|
3 |
size 3226184
|