bharadwajkg
commited on
Commit
•
a67246b
1
Parent(s):
6ac6fa4
End of training
Browse files- README.md +21 -0
- checkpoint-1000/optimizer.bin +3 -0
- checkpoint-1000/pytorch_model.bin +3 -0
- checkpoint-1000/random_states_0.pkl +3 -0
- checkpoint-1000/scaler.pt +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-2000/optimizer.bin +3 -0
- checkpoint-2000/pytorch_model.bin +3 -0
- checkpoint-2000/random_states_0.pkl +3 -0
- checkpoint-2000/scaler.pt +3 -0
- checkpoint-2000/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-4000/optimizer.bin +3 -0
- checkpoint-4000/pytorch_model.bin +3 -0
- checkpoint-4000/random_states_0.pkl +3 -0
- checkpoint-4000/scaler.pt +3 -0
- checkpoint-4000/scheduler.bin +3 -0
- checkpoint-5000/optimizer.bin +3 -0
- checkpoint-5000/pytorch_model.bin +3 -0
- checkpoint-5000/random_states_0.pkl +3 -0
- checkpoint-5000/scaler.pt +3 -0
- checkpoint-5000/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/1690958180.0945485/events.out.tfevents.1690958180.91c35a8d54d2.185.1 +3 -0
- logs/text2image-fine-tune/1690958180.0970595/hparams.yml +51 -0
- logs/text2image-fine-tune/events.out.tfevents.1690958180.91c35a8d54d2.185.0 +3 -0
- pytorch_lora_weights.bin +3 -0
README.md
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
license: creativeml-openrail-m
|
4 |
+
base_model: stabilityai/stable-diffusion-2-1
|
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 - bharadwajkg/finetune-sd2-1-planogram-lora-data5
|
15 |
+
These are LoRA adaption weights for stabilityai/stable-diffusion-2-1. The weights were fine-tuned on the bharadwajkg/planogram-sd-data5 dataset. You can find some example images in the following.
|
16 |
+
|
17 |
+
![img_0](./image_0.png)
|
18 |
+
![img_1](./image_1.png)
|
19 |
+
![img_2](./image_2.png)
|
20 |
+
![img_3](./image_3.png)
|
21 |
+
|
checkpoint-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03ccf62d71610b2bebe3e03032f199bdd7e542ffceac63833b494aa4c9bbe2df
|
3 |
+
size 6853829
|
checkpoint-1000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:059c3f0825e526afb77816eaea34b9c1a3cb51d07372832e6b8ad3177ced2403
|
3 |
+
size 3417037
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d95b002ed823fb9b8f7e607fe08c012d8ce64e79cb3a18e94627054e6c5829a0
|
3 |
+
size 14727
|
checkpoint-1000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
|
3 |
+
size 557
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b0e7c0031b016b4014530d122ec43cabd4876abf9f6428b84177110087b7869
|
3 |
+
size 563
|
checkpoint-2000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:93cb40311cfe04dacf40c666964d26a247cdba110a81cc8ebe34fe4bf941d48b
|
3 |
+
size 6853829
|
checkpoint-2000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ecded346906b12cf057dbf09486d249bba4d43650b993c2c4321ce99b00deb03
|
3 |
+
size 3417037
|
checkpoint-2000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:15fff7ff003c416485eca7143933699e55d62b3d2bade7329f14eb905a51f6a2
|
3 |
+
size 14727
|
checkpoint-2000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd2de9749828adacdf103bf6e9592702bb7067a2c1df27dd62ab38c1eb8c070f
|
3 |
+
size 557
|
checkpoint-2000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:99d048c90a677fd083f3b58a8f8a6bbbeda629e54c112c7166f31c819dd83f52
|
3 |
+
size 563
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:112090402cf1d50af35451e4caaa7f66f57eb97f6da348c108ad1ec35e61500e
|
3 |
+
size 6853829
|
checkpoint-3000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b2a7da25cd111c77acd33f451c126d5bceb2d177916455fa64a3fa3d726fb6c
|
3 |
+
size 3417037
|
checkpoint-3000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7c1583c7a28e7c7e2bdaa5e00ef8154f60fb0865c9f3a6b32c64db0b3901f68f
|
3 |
+
size 14727
|
checkpoint-3000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b1366235b5a98fec34f171d8b5ed88add7c29faac28cf2240061c21ea0737e46
|
3 |
+
size 557
|
checkpoint-3000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8fa6ccd6f463edaff9557fe0e8e9a42eca48dbcae735dbe792c117311113c5b7
|
3 |
+
size 563
|
checkpoint-4000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:af9d2f6768df17e3ea3f6889262a657eaf2f81f88e672bfaad246521f0f40d59
|
3 |
+
size 6853829
|
checkpoint-4000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a1c62cca32a57313cabc2bec017859ae8face0d25ee1327614cc5c226a099021
|
3 |
+
size 3417037
|
checkpoint-4000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f22742c89a5f1e1dc217811574a9e64052e61b74dc202dc799f24e878f0c5368
|
3 |
+
size 14727
|
checkpoint-4000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d98910b5352fb362d925a4731dca8f30464e0497be1a18148b469485af3e7911
|
3 |
+
size 557
|
checkpoint-4000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4226fcb80fe6f26487ccb947096af63ea9aebff0496c3179a3d4ff1f202fa77e
|
3 |
+
size 563
|
checkpoint-5000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ded17a11bf21d7a84c8208fdcc219550ac2d06ea704a8a298ede5ee24189111
|
3 |
+
size 6853829
|
checkpoint-5000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:002baae2f42af501ec18e303a3370332965b72d23edc0c55da9efcb35693973b
|
3 |
+
size 3417037
|
checkpoint-5000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8274ae015640fe11b857721ab10997ee25763eb9df53b3669877ef7729d062b1
|
3 |
+
size 14727
|
checkpoint-5000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7c605152c3bbb0d4e9cadb1c89d89b95be3bc910c6e2dbc675362f7073c41b62
|
3 |
+
size 557
|
checkpoint-5000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:066c7af4333a755f80fdb60a3b2aa69d6b35338dd4854c6abfea08a4afe69606
|
3 |
+
size 563
|
image_0.png
ADDED
image_1.png
ADDED
image_2.png
ADDED
image_3.png
ADDED
logs/text2image-fine-tune/1690958180.0945485/events.out.tfevents.1690958180.91c35a8d54d2.185.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d528e6c37dbb6d74feb9ddde0c0482c92a6c1d2574b0710ab7aba243b4c73295
|
3 |
+
size 2455
|
logs/text2image-fine-tune/1690958180.0970595/hparams.yml
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 1000
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 2
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: bharadwajkg/planogram-sd-data5
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
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: cosine
|
24 |
+
lr_warmup_steps: 0
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 5000
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 67
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: finetune-sd2-1-planogram-lora-data5
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
|
35 |
+
push_to_hub: true
|
36 |
+
random_flip: false
|
37 |
+
rank: 4
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 512
|
40 |
+
resume_from_checkpoint: null
|
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: A rack with 5 Shelf rows, first row 5 yellow Corn Flakes boxes,
|
50 |
+
second row 3 brown Chocos boxes, third row 3 red oats packets, fourth row 4 blue
|
51 |
+
Oats packets, fifth row 4 brown chocolate jars and 3 red Ketchup bottles
|
logs/text2image-fine-tune/events.out.tfevents.1690958180.91c35a8d54d2.185.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e59fd2870233c3d6c4a5dba06881ab595a25f600d32d529804046446637a6ede
|
3 |
+
size 240551865
|
pytorch_lora_weights.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:237ff091703a3d7ac02bbee216de437dce21ce98f80257f4eeb91743355a27c2
|
3 |
+
size 3418843
|