End of training
Browse files- README.md +21 -0
- checkpoint-10000/optimizer.bin +3 -0
- checkpoint-10000/pytorch_model.bin +3 -0
- checkpoint-10000/random_states_0.pkl +3 -0
- checkpoint-10000/scaler.pt +3 -0
- checkpoint-10000/scheduler.bin +3 -0
- checkpoint-3000/optimizer.bin +3 -0
- checkpoint-3000/pytorch_model.bin +3 -0
- checkpoint-3000/random_states_0.pkl +3 -0
- checkpoint-3000/scaler.pt +3 -0
- checkpoint-3000/scheduler.bin +3 -0
- checkpoint-6000/optimizer.bin +3 -0
- checkpoint-6000/pytorch_model.bin +3 -0
- checkpoint-6000/random_states_0.pkl +3 -0
- checkpoint-6000/scaler.pt +3 -0
- checkpoint-6000/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/1701217194.5259678/events.out.tfevents.1701217194.06443603d9c4.27886.1 +3 -0
- logs/text2image-fine-tune/1701217194.5278049/hparams.yml +49 -0
- logs/text2image-fine-tune/1701217652.1088934/events.out.tfevents.1701217652.06443603d9c4.29851.1 +3 -0
- logs/text2image-fine-tune/1701217652.1116586/hparams.yml +49 -0
- logs/text2image-fine-tune/events.out.tfevents.1701217194.06443603d9c4.27886.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1701217652.06443603d9c4.29851.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
license: creativeml-openrail-m
|
4 |
+
base_model: runwayml/stable-diffusion-v1-5
|
5 |
+
tags:
|
6 |
+
- stable-diffusion
|
7 |
+
- stable-diffusion-diffusers
|
8 |
+
- text-to-image
|
9 |
+
- diffusers
|
10 |
+
- lora
|
11 |
+
inference: true
|
12 |
+
---
|
13 |
+
|
14 |
+
# LoRA text2image fine-tuning - rrw23/train8
|
15 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the pcuenq/oxford-pets dataset. You can find some example images in the following.
|
16 |
+
|
17 |
+

|
18 |
+

|
19 |
+

|
20 |
+

|
21 |
+
|
checkpoint-10000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d4e7bda02d46444a880eb58dcd8f0199954a729d74d416c51c11d6e0da85b87f
|
3 |
+
size 25724538
|
checkpoint-10000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a9df23b33592ddda5961b7a7296c1b86c76683276cbda5bfeda5f9f632ae03e6
|
3 |
+
size 12852618
|
checkpoint-10000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bd6ab3721e60f593ad2c5ae05f453422887864fbc6685eb0787122168408a0d4
|
3 |
+
size 14408
|
checkpoint-10000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d49367aa0f19981d0484e3ef5c1b009e6bdd8e5c0d8ef08f2f8d235b9f843816
|
3 |
+
size 988
|
checkpoint-10000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:80d1ae15c178ebb7384cbccab125f55970b378170012f20a97068abd32ad69dc
|
3 |
+
size 1000
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c3bd7da85520ccf18ef1d266399801fdf0c8abc79e5a58fbdc942cc3cb29ed41
|
3 |
+
size 25724538
|
checkpoint-3000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3d932025d743ba5949b46a24817489fbb0fcf865d91c09e1a525b269b6dc02ae
|
3 |
+
size 12852618
|
checkpoint-3000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:386a047dcd3f94ac9f6f937c14848ea3a03edbc57f37cd7c1f430417b1735c39
|
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:9e31a2548418a1ef5784752739e90341510cf150ab5e3aba4f3208c787d1d6bb
|
3 |
+
size 1000
|
checkpoint-6000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:53cf6a2421c2c52d6523075ced95185bb8fc064100495fb6377f4ad4d87d53ba
|
3 |
+
size 25724538
|
checkpoint-6000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d9008ad6980a91e3dffe1e2430ff0b56f983edc1a823f232c98eadc8017df20d
|
3 |
+
size 12852618
|
checkpoint-6000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0542b06fd5212985cee79c0ff01e6695583985d1d6a06e495bdd7ac4326e563d
|
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:4e5539e076d6ec3a359fdc4aa95373fa97869b5c13ce584bf7ff030b3b14d67b
|
3 |
+
size 1000
|
image_0.png
ADDED
![]() |
image_1.png
ADDED
![]() |
image_2.png
ADDED
![]() |
image_3.png
ADDED
![]() |
logs/text2image-fine-tune/1701217194.5259678/events.out.tfevents.1701217194.06443603d9c4.27886.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5835f9fb066e36d7063223fcc99e2a8d2a16c993468580aa1219fd683687c64f
|
3 |
+
size 2359
|
logs/text2image-fine-tune/1701217194.5278049/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 |
+
cache_dir: null
|
7 |
+
caption_column: label
|
8 |
+
center_crop: true
|
9 |
+
checkpointing_steps: 500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 8
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: pcuenq/oxford-pets
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: train8
|
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: 10001
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 22
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: /content/weights/
|
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: 16
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 256
|
40 |
+
resume_from_checkpoint: ../../../weights/checkpoint-10000
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 1337
|
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: siamese cat on grass
|
logs/text2image-fine-tune/1701217652.1088934/events.out.tfevents.1701217652.06443603d9c4.29851.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2892ae082085eda3ba18419f46cfbc1b4f99686adaed248bd858d56cb3c06fd2
|
3 |
+
size 2359
|
logs/text2image-fine-tune/1701217652.1116586/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 |
+
cache_dir: null
|
7 |
+
caption_column: label
|
8 |
+
center_crop: true
|
9 |
+
checkpointing_steps: 500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 8
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: pcuenq/oxford-pets
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: train8
|
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: 10001
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 22
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: /content/weights/
|
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: 16
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 256
|
40 |
+
resume_from_checkpoint: ../../../weights/checkpoint-10000
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 1337
|
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: siamese cat on grass
|
logs/text2image-fine-tune/events.out.tfevents.1701217194.06443603d9c4.27886.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:26f2861a90122572b8c4e0fd8812c83e6b87231f2735c2fc268880a8061348a9
|
3 |
+
size 1988377
|
logs/text2image-fine-tune/events.out.tfevents.1701217652.06443603d9c4.29851.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:42f3e78b5211a4a591dfbf9431fbbbbbd33c01d0abd0fa0738b58378ccdf23e0
|
3 |
+
size 1986697
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03ab5add46d981b4788876b8263e8a8a1f6ba90a8ca81679193f4f799baca4ed
|
3 |
+
size 12794104
|