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/1701212165.9925437/events.out.tfevents.1701212165.06443603d9c4.6504.1 +3 -0
- logs/text2image-fine-tune/1701212165.995505/hparams.yml +50 -0
- logs/text2image-fine-tune/1701212584.124752/events.out.tfevents.1701212584.06443603d9c4.8398.1 +3 -0
- logs/text2image-fine-tune/1701212584.1265583/hparams.yml +49 -0
- logs/text2image-fine-tune/events.out.tfevents.1701212165.06443603d9c4.6504.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1701212584.06443603d9c4.8398.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/train9
|
| 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:9dd20d0d94d185ca7c29cce3759dc551e6c667d29c42488bb2a463b16ab9fef9
|
| 3 |
+
size 6592122
|
checkpoint-10000/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:18fa870fbfc459c81e43be8d6e91ca88719497951d3ad679a11d6827fe4c03e3
|
| 3 |
+
size 3286410
|
checkpoint-10000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b6e5f46d01056cd27f9a1bf1131b78ce6f2d997ceeb7df8d6c5d97f0466f358
|
| 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:571da3e9cccdb5470dc3c5fb98bdc1e719ff4a5231a78c5114f03e01a05a2424
|
| 3 |
+
size 1000
|
checkpoint-3000/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:be26747a19826c5c6d1bd1ace7b550ef6f7c86ef484618ab7104a085ab49ad3f
|
| 3 |
+
size 6592122
|
checkpoint-3000/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e15cf16ad8bb4d838747bf981138652339c76d35f05a422966bf9401d765e27
|
| 3 |
+
size 3286410
|
checkpoint-3000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:87cdec6688a436e3a6de8fd88a0b155cd7f8005eb83394bae2837a848d5b5826
|
| 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:d14a229fdc74134f115fc67bd1633c2dae663b2ee00e650109eaa0fe4d9964a6
|
| 3 |
+
size 1000
|
checkpoint-6000/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e545e5d0e09542c078a3aa86772a1c533931c799e1f1b55c56fef4235589eadf
|
| 3 |
+
size 6592122
|
checkpoint-6000/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:88ba661b7c07e2d7884cd438884c7a4b53476f4e98860559a31dd62f1c790a14
|
| 3 |
+
size 3286410
|
checkpoint-6000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fccd7ac8d106b0a4d26d422150a0350f98712dea0b7f7c5a82686bc0282f2a20
|
| 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:aa32746448fbdcf0f58ce40566d40f767617ab9d710dca78c9c8c119899a4922
|
| 3 |
+
size 1000
|
image_0.png
ADDED
|
image_1.png
ADDED
|
image_2.png
ADDED
|
image_3.png
ADDED
|
logs/text2image-fine-tune/1701212165.9925437/events.out.tfevents.1701212165.06443603d9c4.6504.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:15bced714186db3b9997a33e153bdeabc74136e0effd11e6f6c964a7d44129cd
|
| 3 |
+
size 2372
|
logs/text2image-fine-tune/1701212165.995505/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: 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: train9
|
| 18 |
+
hub_token: null
|
| 19 |
+
image_column: image
|
| 20 |
+
learning_rate: 1.0e-05
|
| 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: 7
|
| 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: 4
|
| 38 |
+
report_to: tensorboard
|
| 39 |
+
resolution: 256
|
| 40 |
+
resume_from_checkpoint: latest
|
| 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 |
+
val_batch_size: 1
|
| 49 |
+
validation_epochs: 1
|
| 50 |
+
validation_prompt: siamese cat
|
logs/text2image-fine-tune/1701212584.124752/events.out.tfevents.1701212584.06443603d9c4.8398.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:25d5f789a9d10d41abb6b602a741cbc4582cd41bb584615f33efa4c4011fd6dc
|
| 3 |
+
size 2323
|
logs/text2image-fine-tune/1701212584.1265583/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: train9
|
| 18 |
+
hub_token: null
|
| 19 |
+
image_column: image
|
| 20 |
+
learning_rate: 1.0e-05
|
| 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: 4
|
| 38 |
+
report_to: tensorboard
|
| 39 |
+
resolution: 256
|
| 40 |
+
resume_from_checkpoint: latest
|
| 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
|
logs/text2image-fine-tune/events.out.tfevents.1701212165.06443603d9c4.6504.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c33b5b729f32a79c8d066aca0c650f283f40b5dda3f7e4c204b832d057ea1fb
|
| 3 |
+
size 184
|
logs/text2image-fine-tune/events.out.tfevents.1701212584.06443603d9c4.8398.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c000e5cf9df67277d7f05b509d8fbd794d587c92d704a15785c883ddd8460887
|
| 3 |
+
size 1429641
|
pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ee0c7a44996933c9a954ea945437b03c742a15c7d99063c73332c553f2dc1bf
|
| 3 |
+
size 3227336
|