End of training
Browse files- README.md +21 -0
- checkpoint-25/model.safetensors +3 -0
- checkpoint-25/optimizer.bin +3 -0
- checkpoint-25/pytorch_lora_weights.safetensors +3 -0
- checkpoint-25/random_states_0.pkl +3 -0
- checkpoint-25/scheduler.bin +3 -0
- checkpoint-2500/model.safetensors +3 -0
- checkpoint-2500/optimizer.bin +3 -0
- checkpoint-2500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2500/random_states_0.pkl +3 -0
- checkpoint-2500/scheduler.bin +3 -0
- checkpoint-2750/model.safetensors +3 -0
- checkpoint-2750/optimizer.bin +3 -0
- checkpoint-2750/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2750/random_states_0.pkl +3 -0
- checkpoint-2750/scheduler.bin +3 -0
- checkpoint-5000/model.safetensors +3 -0
- checkpoint-5500/model.safetensors +3 -0
- checkpoint-5500/optimizer.bin +3 -0
- checkpoint-5500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-5500/random_states_0.pkl +3 -0
- checkpoint-5500/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/1706847006.4238968/events.out.tfevents.1706847006.ip-172-16-22-220.ec2.internal.30055.1 +3 -0
- logs/text2image-fine-tune/1706847006.4256144/hparams.yml +52 -0
- logs/text2image-fine-tune/1706866093.104948/events.out.tfevents.1706866093.ip-172-16-17-134.ec2.internal.14016.1 +3 -0
- logs/text2image-fine-tune/1706866093.1065829/hparams.yml +52 -0
- logs/text2image-fine-tune/1706866705.6231158/events.out.tfevents.1706866705.ip-172-16-17-134.ec2.internal.22288.1 +3 -0
- logs/text2image-fine-tune/1706866705.6247342/hparams.yml +52 -0
- logs/text2image-fine-tune/1706866734.1540847/events.out.tfevents.1706866734.ip-172-16-17-134.ec2.internal.22718.1 +3 -0
- logs/text2image-fine-tune/1706866734.1557229/hparams.yml +52 -0
- logs/text2image-fine-tune/1706866932.980362/events.out.tfevents.1706866932.ip-172-16-17-134.ec2.internal.25214.1 +3 -0
- logs/text2image-fine-tune/1706866932.9819658/hparams.yml +52 -0
- logs/text2image-fine-tune/events.out.tfevents.1706847006.ip-172-16-22-220.ec2.internal.30055.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1706863514.ip-172-16-22-220.ec2.internal.10732.0 +0 -0
- logs/text2image-fine-tune/events.out.tfevents.1706863542.ip-172-16-22-220.ec2.internal.11114.0 +0 -0
- logs/text2image-fine-tune/events.out.tfevents.1706863847.ip-172-16-22-220.ec2.internal.15052.0 +0 -0
- logs/text2image-fine-tune/events.out.tfevents.1706866093.ip-172-16-17-134.ec2.internal.14016.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1706866705.ip-172-16-17-134.ec2.internal.22288.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1706866734.ip-172-16-17-134.ec2.internal.22718.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1706866932.ip-172-16-17-134.ec2.internal.25214.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 - diksha13/arrivae-subset-top-21-sage
|
15 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the diksha13/top-21-subset 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-25/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6e6d9ca2e7a8e237736290cf2cebc9587cf599dfc8f20e7bccaceced99b944d9
|
3 |
+
size 3441397096
|
checkpoint-25/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:60f69172fa94967ddac7676787f69c2f36eef0435932ba3189990c1ab0d4b341
|
3 |
+
size 6592122
|
checkpoint-25/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fec98147adaed8bff031bd9afdc3a71b048df891c8d0089679f68ec20bf415e4
|
3 |
+
size 3226184
|
checkpoint-25/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e67a5335c5e749b63e08fb1c9a30bda786655523f47b5684ecdb407c632021e1
|
3 |
+
size 14408
|
checkpoint-25/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec9af3bb30460db2c3db1af582f8239ab615318f7c9eb8d6d86b94c218b0ed65
|
3 |
+
size 1000
|
checkpoint-2500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:52b4e54efde21828f812c74b5e1da01a5f17f55368104ef1f7904e306ba8f142
|
3 |
+
size 3441397096
|
checkpoint-2500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eeb88d1ff19386f42fb192910d3b860cb7f7d89ee19f3fc8c384c9e33be92bbf
|
3 |
+
size 6592122
|
checkpoint-2500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f8be654f6a9efed17858c5bcfb08ef24109293180fdacdf17a7939eee74858f6
|
3 |
+
size 3226184
|
checkpoint-2500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a5ac1a3f0e19dcf9f8ff126574a607b38ef4a2ccc85c76052b2d64c4b0b28d61
|
3 |
+
size 14408
|
checkpoint-2500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:550e89594a65aa85e261bf46f6d025a07da81177c94b2438db504a01644a638d
|
3 |
+
size 1000
|
checkpoint-2750/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e94c9cb0e04ec0612f676598d0cead0ebeec513c1727dc1b6666a76c5cbc2fed
|
3 |
+
size 3441397096
|
checkpoint-2750/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23768f58c03afdda6853f7bd8c058c8275c1d5de74f30bbc643171df42f604cf
|
3 |
+
size 6592122
|
checkpoint-2750/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f73aad28ccc9645bf69d1ad060034911b83904e1c4915fefac2adf5b8d0d0600
|
3 |
+
size 3226184
|
checkpoint-2750/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5bbca2177294bd4786438690496e9300f9c17eea9239eef7c7a9b59abd172372
|
3 |
+
size 14408
|
checkpoint-2750/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2f3aa506e3a544e48fc4957e0d56bfc6a571b9b1c3b208179b7c8c4282238fc3
|
3 |
+
size 1000
|
checkpoint-5000/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e91b83b14dc1ea6edf3b184581943224c93a59fb7c2367ad7843ffb17ae08a8e
|
3 |
+
size 1207685120
|
checkpoint-5500/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:65718be265de4bb21824b1017c602c756407488a7cc48eeaa5813acc0e092aa8
|
3 |
+
size 3441397096
|
checkpoint-5500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:08eb170b66710f6425e673c97053bbebffbc04698d941e78692f2ab5824bcdc3
|
3 |
+
size 6592122
|
checkpoint-5500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:98d50a4ca70e4cff06a63a6df42f559b3bd7debc8f0bc65e91c3a8552ff1f786
|
3 |
+
size 3226184
|
checkpoint-5500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:94eba6ce868dcbffd48e73feca49e11138a1660906d458a9ccfb44c31ba0a760
|
3 |
+
size 14408
|
checkpoint-5500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:85171da70725c3e5415488e11df475665ed8ee557b4f9d3d3f69763589e5ba6b
|
3 |
+
size 1000
|
image_0.png
ADDED
![]() |
image_1.png
ADDED
![]() |
image_2.png
ADDED
![]() |
image_3.png
ADDED
![]() |
logs/text2image-fine-tune/1706847006.4238968/events.out.tfevents.1706847006.ip-172-16-22-220.ec2.internal.30055.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e01a8b0421b7697c6dec1bee114fa690e58fcfda1f1d38269cdc64b98790bc2
|
3 |
+
size 2474
|
logs/text2image-fine-tune/1706847006.4256144/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 2500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 4
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: diksha13/top-21-subset
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: arrivae-subset-top-21-sage
|
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: 36
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: 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: null
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 76865768
|
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 front angle view of a ranger high back office chair having a
|
50 |
+
bright yellow back rest, black seats, wheets, and handles. It is facing the camera,
|
51 |
+
has five chrome wheels. high definition, upscaled, photorealistic
|
52 |
+
variant: null
|
logs/text2image-fine-tune/1706866093.104948/events.out.tfevents.1706866093.ip-172-16-17-134.ec2.internal.14016.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5de3427ef1d7f2e14bdcc02fba192e9d7691e230b487316a9494535e9470020a
|
3 |
+
size 2474
|
logs/text2image-fine-tune/1706866093.1065829/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 2500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 4
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: diksha13/top-21-subset
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: arrivae-subset-top-21-sage
|
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: 50
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 1
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
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: 76865768
|
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 front angle view of a ranger high back office chair having a
|
50 |
+
bright yellow back rest, black seats, wheets, and handles. It is facing the camera,
|
51 |
+
has five chrome wheels. high definition, upscaled, photorealistic
|
52 |
+
variant: null
|
logs/text2image-fine-tune/1706866705.6231158/events.out.tfevents.1706866705.ip-172-16-17-134.ec2.internal.22288.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bc306d7a5f14a3dc9a5cbb00b39082fb453a5124813cb8af16c8b6d1f499912d
|
3 |
+
size 2476
|
logs/text2image-fine-tune/1706866705.6247342/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 2500
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 4
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: diksha13/top-21-subset
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: arrivae-subset-top-21-sage
|
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: 36
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: ./model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
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: 76865768
|
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 front angle view of a ranger high back office chair having a
|
50 |
+
bright yellow back rest, black seats, wheets, and handles. It is facing the camera,
|
51 |
+
has five chrome wheels. high definition, upscaled, photorealistic
|
52 |
+
variant: null
|
logs/text2image-fine-tune/1706866734.1540847/events.out.tfevents.1706866734.ip-172-16-17-134.ec2.internal.22718.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e6c1cd01d3fc0dc12b120c16238dc89cd253d1d3a683a4a6c4f2121ce56dc08e
|
3 |
+
size 2476
|
logs/text2image-fine-tune/1706866734.1557229/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 25
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 4
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: diksha13/top-21-subset
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: arrivae-subset-top-21-sage
|
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: 50
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 1
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: ./model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
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: 76865768
|
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 front angle view of a ranger high back office chair having a
|
50 |
+
bright yellow back rest, black seats, wheets, and handles. It is facing the camera,
|
51 |
+
has five chrome wheels. high definition, upscaled, photorealistic
|
52 |
+
variant: null
|
logs/text2image-fine-tune/1706866932.980362/events.out.tfevents.1706866932.ip-172-16-17-134.ec2.internal.25214.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e9170a56c110761ce055db39583bfb795d9e8a0ae939a966efe77d7ca7fcb4e2
|
3 |
+
size 2476
|
logs/text2image-fine-tune/1706866932.9819658/hparams.yml
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: 2750
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 4
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: diksha13/top-21-subset
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: arrivae-subset-top-21-sage
|
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: 5500
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 39
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: ./model
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
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: 76865768
|
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 front angle view of a ranger high back office chair having a
|
50 |
+
bright yellow back rest, black seats, wheets, and handles. It is facing the camera,
|
51 |
+
has five chrome wheels. high definition, upscaled, photorealistic
|
52 |
+
variant: null
|
logs/text2image-fine-tune/events.out.tfevents.1706847006.ip-172-16-22-220.ec2.internal.30055.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7cdc634c27596bd86cb4266df071650f2a1660d047031e22e25032e7992f8811
|
3 |
+
size 60998269
|
logs/text2image-fine-tune/events.out.tfevents.1706863514.ip-172-16-22-220.ec2.internal.10732.0
ADDED
File without changes
|
logs/text2image-fine-tune/events.out.tfevents.1706863542.ip-172-16-22-220.ec2.internal.11114.0
ADDED
File without changes
|
logs/text2image-fine-tune/events.out.tfevents.1706863847.ip-172-16-22-220.ec2.internal.15052.0
ADDED
File without changes
|
logs/text2image-fine-tune/events.out.tfevents.1706866093.ip-172-16-17-134.ec2.internal.14016.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8c30920a7aa2fff72023699a8db81040e8dfe6044fbc52034ba8eba97aebd735
|
3 |
+
size 3548652
|
logs/text2image-fine-tune/events.out.tfevents.1706866705.ip-172-16-17-134.ec2.internal.22288.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c8b4f5f1198c3a4cd4332d243609c1dc8e5cc90e4cd27782fd3493fc60fa4ed9
|
3 |
+
size 280
|
logs/text2image-fine-tune/events.out.tfevents.1706866734.ip-172-16-17-134.ec2.internal.22718.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:55eb8fc4d98dfc1e73d00d6b44f42612f3eee753141a3006f9658db5a96cee56
|
3 |
+
size 2488
|
logs/text2image-fine-tune/events.out.tfevents.1706866932.ip-172-16-17-134.ec2.internal.25214.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0379852501203adc926c8006032325b1c7fc98809c2d98b06be6bb011a764f17
|
3 |
+
size 65093531
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:98d50a4ca70e4cff06a63a6df42f559b3bd7debc8f0bc65e91c3a8552ff1f786
|
3 |
+
size 3226184
|