Commit ·
84f9726
1
Parent(s): 71a27af
End of training
Browse files- README.md +19 -1
- checkpoint-125/optimizer.bin +3 -0
- checkpoint-125/pytorch_model.bin +3 -0
- checkpoint-125/random_states_0.pkl +3 -0
- checkpoint-125/scaler.pt +3 -0
- checkpoint-125/scheduler.bin +3 -0
- checkpoint-130/optimizer.bin +3 -0
- checkpoint-130/pytorch_model.bin +3 -0
- checkpoint-130/random_states_0.pkl +3 -0
- checkpoint-130/scaler.pt +3 -0
- checkpoint-130/scheduler.bin +3 -0
- checkpoint-135/optimizer.bin +3 -0
- checkpoint-135/pytorch_model.bin +3 -0
- checkpoint-135/random_states_0.pkl +3 -0
- checkpoint-135/scaler.pt +3 -0
- checkpoint-135/scheduler.bin +3 -0
- checkpoint-140/optimizer.bin +3 -0
- checkpoint-140/pytorch_model.bin +3 -0
- checkpoint-140/random_states_0.pkl +3 -0
- checkpoint-140/scaler.pt +3 -0
- checkpoint-140/scheduler.bin +3 -0
- checkpoint-145/optimizer.bin +3 -0
- checkpoint-145/pytorch_model.bin +3 -0
- checkpoint-145/random_states_0.pkl +3 -0
- checkpoint-145/scaler.pt +3 -0
- checkpoint-145/scheduler.bin +3 -0
- checkpoint-150/optimizer.bin +3 -0
- checkpoint-150/pytorch_model.bin +3 -0
- checkpoint-150/random_states_0.pkl +3 -0
- checkpoint-150/scaler.pt +3 -0
- checkpoint-150/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/1686552744.952945/events.out.tfevents.1686552744.DEXTER02.9396.1 +3 -0
- logs/text2image-fine-tune/1686552744.952945/hparams.yml +50 -0
- logs/text2image-fine-tune/events.out.tfevents.1686552744.DEXTER02.9396.0 +3 -0
- pytorch_lora_weights.bin +3 -0
README.md
CHANGED
|
@@ -1,3 +1,21 @@
|
|
|
|
|
| 1 |
---
|
| 2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 - JennnDexter/pokemon-lora
|
| 15 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the lambdalabs/pokemon-blip-captions dataset. You can find some example images in the following.
|
| 16 |
+
|
| 17 |
+

|
| 18 |
+

|
| 19 |
+

|
| 20 |
+

|
| 21 |
+
|
checkpoint-125/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0751a2b28c8f18db8701931ca0139bf0d7a71bab317cf2a0d1a04664f61ef2bd
|
| 3 |
+
size 6591685
|
checkpoint-125/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cdeac4db715d34b0d4c702cd1cf2ef41d1a4fdd7f1d7ed4bdf61339cdaea9a0f
|
| 3 |
+
size 3285965
|
checkpoint-125/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a80b770bd22b2925554839d43c8b3f3909c380e37e08ae666e21bfefc1f116e
|
| 3 |
+
size 14727
|
checkpoint-125/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1e00e30d6c8604ed94f887e4abb61441502990693d51d0b7cc56e46dad4cc26c
|
| 3 |
+
size 557
|
checkpoint-125/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:078f50bb294b5123f62d7ce6b98d9d4c0c8f7151959fad3797cdffed52334ce3
|
| 3 |
+
size 563
|
checkpoint-130/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b60ff6aed65f43bcd4819dc0dbf0cc385aef224dbb67f1d8984b60a6ac65927e
|
| 3 |
+
size 6591685
|
checkpoint-130/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3136c71db1f9fe44acfd52d4c42ed4860b1dda4779e6cd40a649b487445b883d
|
| 3 |
+
size 3285965
|
checkpoint-130/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b27e0f59d0321c74ed45a362867ed5a3bdabab3095632a9768aa22ae84b80581
|
| 3 |
+
size 14727
|
checkpoint-130/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31553f62e0460655834ed8ea10094ca204e2bcbb40f0acf86c961cf52acab220
|
| 3 |
+
size 557
|
checkpoint-130/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8f646393d1df851534585d5ea157fbca2ef71ca4585339b727395c3956c44c4
|
| 3 |
+
size 563
|
checkpoint-135/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:15018cc9d397f4b146fa5ec9085881a111f349a779f7d9f79a5c43ab0b9ea4e7
|
| 3 |
+
size 6591685
|
checkpoint-135/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5ed633663401d925c25badf848d8a41c421051e898c6e1e6a6d867afe88c1cc
|
| 3 |
+
size 3285965
|
checkpoint-135/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:23f841f56b5d8ee99b5a8e442b8954cd2d772ca2d0fd4edda23cbec06036c4bb
|
| 3 |
+
size 14727
|
checkpoint-135/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1396f7cb860da886b0e6a2ed1cde6a4556f6d5fcda50c7453c5fd66d7afbb50e
|
| 3 |
+
size 557
|
checkpoint-135/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8dc73bde0b308fe6cb967c96878e9c079489cff09d3cf5bec76fb238deb8309a
|
| 3 |
+
size 563
|
checkpoint-140/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:47afd89680e9400f6017fcceb0a062bb6ef7a6be114aebfebb6561e55dde4b2d
|
| 3 |
+
size 6591685
|
checkpoint-140/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:34b6c1bb508e6865478079db5eb89db5feb609f027bb4c864abb13f7db49525e
|
| 3 |
+
size 3285965
|
checkpoint-140/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:469bc28484fa11b27f9cec31caa4b8b6b25087ee6e065760f1c058f578e3632e
|
| 3 |
+
size 14727
|
checkpoint-140/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7dbbfbece2fede2a53cc80ae296080e69d39722b90afa9ebeb253ed2579b6c4
|
| 3 |
+
size 557
|
checkpoint-140/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a8da5664d275174fc31a1ae0ef212e7ba2193a332df8600444dc564c019b540
|
| 3 |
+
size 563
|
checkpoint-145/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:69f3aa66de215cbc33e62f03acfe21629ff7fc50a2f1fe1fc83c6fe8bb48de39
|
| 3 |
+
size 6591685
|
checkpoint-145/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99b641e2bfe3256d6e9ded22c04afd3561157543553d8d784514c366503cb0bc
|
| 3 |
+
size 3285965
|
checkpoint-145/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3dcb2a3a4357b5d7fa38e76460d59ed848dc349655425e6fdedcbeda5007db23
|
| 3 |
+
size 14727
|
checkpoint-145/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:34be8bcc426a463091c9197a0f37bcadfbbf5f13cb851d1cc2951965799b123f
|
| 3 |
+
size 557
|
checkpoint-145/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1dd07fcb2e365ddd8898c31c99e5fbce20379cb43454e15d05560d1f68bc7a08
|
| 3 |
+
size 563
|
checkpoint-150/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:794bf9df7783993693ce88686a4b9bc28ffd295e9d99e3805dbb26a200937ff7
|
| 3 |
+
size 6591685
|
checkpoint-150/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ac6bb2deada58ab993a29f9686227429d44db1a80985a59377912ad417bf6902
|
| 3 |
+
size 3285965
|
checkpoint-150/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:059b02656359b7826a0996caeeb476eca97ae9d8c8384b2bc865e3d413967036
|
| 3 |
+
size 14727
|
checkpoint-150/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c51c8702e03bc24d0566e4b37e893acfcfe0af87dcc3883f22471d07682d7ac5
|
| 3 |
+
size 557
|
checkpoint-150/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f98d46901ad4874fceeb76bb128871c1d90eb6764b534b49e68e6ea66d4314d
|
| 3 |
+
size 563
|
image_0.png
ADDED
|
image_1.png
ADDED
|
image_2.png
ADDED
|
image_3.png
ADDED
|
logs/text2image-fine-tune/1686552744.952945/events.out.tfevents.1686552744.DEXTER02.9396.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f4e30af2cd9c001799e647276f457f6b38115c6c8ac1896cb23abd9aa46f89b
|
| 3 |
+
size 2420
|
logs/text2image-fine-tune/1686552744.952945/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: 5
|
| 10 |
+
checkpoints_total_limit: null
|
| 11 |
+
dataloader_num_workers: 0
|
| 12 |
+
dataset_config_name: null
|
| 13 |
+
dataset_name: lambdalabs/pokemon-blip-captions
|
| 14 |
+
enable_xformers_memory_efficient_attention: false
|
| 15 |
+
gradient_accumulation_steps: 4
|
| 16 |
+
gradient_checkpointing: false
|
| 17 |
+
hub_model_id: pokemon-lora
|
| 18 |
+
hub_private_repo: false
|
| 19 |
+
hub_token: null
|
| 20 |
+
image_column: image
|
| 21 |
+
learning_rate: 0.0001
|
| 22 |
+
local_rank: -1
|
| 23 |
+
logging_dir: logs
|
| 24 |
+
lr_scheduler: cosine
|
| 25 |
+
lr_warmup_steps: 0
|
| 26 |
+
max_grad_norm: 1
|
| 27 |
+
max_train_samples: null
|
| 28 |
+
max_train_steps: 150
|
| 29 |
+
mixed_precision: fp16
|
| 30 |
+
noise_offset: 0
|
| 31 |
+
num_train_epochs: 1
|
| 32 |
+
num_validation_images: 4
|
| 33 |
+
output_dir: pokemon-lora
|
| 34 |
+
overwrite_output_dir: true
|
| 35 |
+
prediction_type: null
|
| 36 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
| 37 |
+
push_to_hub: true
|
| 38 |
+
random_flip: false
|
| 39 |
+
report_to: tensorboard
|
| 40 |
+
resolution: 512
|
| 41 |
+
resume_from_checkpoint: null
|
| 42 |
+
revision: null
|
| 43 |
+
scale_lr: false
|
| 44 |
+
seed: 1337
|
| 45 |
+
snr_gamma: null
|
| 46 |
+
train_batch_size: 1
|
| 47 |
+
train_data_dir: null
|
| 48 |
+
use_8bit_adam: false
|
| 49 |
+
validation_epochs: 1
|
| 50 |
+
validation_prompt: A pokemon with blue eyes.
|
logs/text2image-fine-tune/events.out.tfevents.1686552744.DEXTER02.9396.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5936787b42c277838971865b211b1c8d55a476c7b766d5c545bdc503d53504d2
|
| 3 |
+
size 1616323
|
pytorch_lora_weights.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d688f991b315600a751f5b753b92981ea04eeefc839d10721243ee61fa94d567
|
| 3 |
+
size 3287771
|