MaxReynolds
commited on
Commit
•
49187ff
1
Parent(s):
d73e210
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-1500/optimizer.bin +3 -0
- checkpoint-1500/pytorch_model.bin +3 -0
- checkpoint-1500/random_states_0.pkl +3 -0
- checkpoint-1500/scaler.pt +3 -0
- checkpoint-1500/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-2500/optimizer.bin +3 -0
- checkpoint-2500/pytorch_model.bin +3 -0
- checkpoint-2500/random_states_0.pkl +3 -0
- checkpoint-2500/scaler.pt +3 -0
- checkpoint-2500/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-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_model.bin +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scaler.pt +3 -0
- checkpoint-500/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/1696536308.1885316/events.out.tfevents.1696536308.092857a7bc4b.15301.1 +3 -0
- logs/text2image-fine-tune/1696536308.190128/hparams.yml +49 -0
- logs/text2image-fine-tune/1696545071.745234/events.out.tfevents.1696545071.092857a7bc4b.109216.1 +3 -0
- logs/text2image-fine-tune/1696545071.74736/hparams.yml +50 -0
- logs/text2image-fine-tune/events.out.tfevents.1696536308.092857a7bc4b.15301.0 +3 -0
- logs/text2image-fine-tune/events.out.tfevents.1696545071.092857a7bc4b.109216.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: CompVis/stable-diffusion-v1-2
|
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 - MaxReynolds/SouderRocketLauncherNet
|
15 |
+
These are LoRA adaption weights for CompVis/stable-diffusion-v1-2. The weights were fine-tuned on the MaxReynolds/Lee_Souder_RocketLauncher 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:26cc1d87674390b063b12ee3665c799ca43b0f5bea534bc119e7456eae05497b
|
3 |
+
size 6592122
|
checkpoint-1000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bb9fef38e0dbd907f9106488f65c29aa4c5173ae97042fe1d8e123f09a32c46b
|
3 |
+
size 3286410
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:76288b28b4322c17e1fa4a0693e6fc84af6baa014145d7b4800af9d3c29d7528
|
3 |
+
size 14408
|
checkpoint-1000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d8fdcd0311eba9854fff738038ed4c1a269832665b4d88ba4e4e3d02a1a7e0e
|
3 |
+
size 988
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b5d9e05d4cdf29c2fa62a23d3f6adb8f83a1fbc0731ed38f18c48d5f9e248f89
|
3 |
+
size 1000
|
checkpoint-1500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4e43d63e8a6ea4c355561628c926eb2f92e419ceac604fbf38929f44257d845d
|
3 |
+
size 6592122
|
checkpoint-1500/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b402d2d0080d3768f2dd7c4de456dbea6b622b64b8f71a387ad3ef72d99d9574
|
3 |
+
size 3286410
|
checkpoint-1500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:81a4e49cbb01080b0243bb3a0293d6194ab6d52163099586e25201e7c4b30ed0
|
3 |
+
size 14408
|
checkpoint-1500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c2a4b6e0ef05ca249f2d6b7f9f3ad1dff81e55842a962df795bb9740c17c8e92
|
3 |
+
size 988
|
checkpoint-1500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3f23c23385a1cff44e2dd4752cf188d40380ee94561a4ae81794a29a5a19447b
|
3 |
+
size 1000
|
checkpoint-2000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:53af2d56b13e6f483d4e5a319cb1f451d743d266eec3219b5b26c03bd6ed4d53
|
3 |
+
size 6592122
|
checkpoint-2000/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:32c5b56511c4ba486b7dc432e16ed01c8ae251c55e0ed15c61e0aee153fc8211
|
3 |
+
size 3286410
|
checkpoint-2000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:025a77643af1750500a7fd94003cdc02dee05a2945fb1a9378a3eec82e61add7
|
3 |
+
size 14408
|
checkpoint-2000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3c50a9cebe5d66d453d25b140738bff479749ac03e0a43597d8776bc22f6ed0c
|
3 |
+
size 988
|
checkpoint-2000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e04519463d3667dd14d4c697b848d64d915d1101dbf0ed4e8acd9d22773e5470
|
3 |
+
size 1000
|
checkpoint-2500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d21dd77fd2fd32aa3ea028dda87f802d9f225716933bc5c1046a2fd7369129dc
|
3 |
+
size 6592122
|
checkpoint-2500/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:82e3d682863b19991331ba4ac65aec5d50452d9986b913aa933f942c8c8d3451
|
3 |
+
size 3286410
|
checkpoint-2500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4268dacf55006112b8f17daf5fb748fc5bd2aa7cf467343156e4258528b9b0c8
|
3 |
+
size 14408
|
checkpoint-2500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:48e2d97f563bb838328076a1666504681962151a3975a2f064be3a03e6500740
|
3 |
+
size 988
|
checkpoint-2500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f543ae6f3c3aea2020c464822ce19050764a497581df1fae7a798b739acbc204
|
3 |
+
size 1000
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9b49870ac38a3039bbe7d874e493a16851cab7bbdb367b6e28ac8a38ac24a7a5
|
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:a13d9bbfd0f4f60dd53dc424078c5c0485bf67a0fa2e977a11713f88871f1ab4
|
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:c50871abd56cdb7ee5c650286b6f802c4574d7f464c55846d2caa86094e32ee5
|
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:89df8bcca59003290e3a854d22c2892d49d485440014cbb479fc1d4c852903d2
|
3 |
+
size 1000
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2db95c198dd801e24a22ab4e457bf68bbad7cf7653fd2101d7fc028a2d41f0fe
|
3 |
+
size 6592122
|
checkpoint-500/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:856127665cab1201d0686abd8529ba0b603f1aead7ded2708b2c8d3c5ac2359d
|
3 |
+
size 3286410
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c6524782421c6fc926a54b5e7c5190486746e54242d1b2ef128466a2adfca34f
|
3 |
+
size 14408
|
checkpoint-500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:18b984273ea2d45b7ffb1d047bb359d93111e41fcad70d16a1b453fd38f72636
|
3 |
+
size 988
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d70350f67cce0e5059c579ddd0239f8b6b5e77b086d8aedf3623af021a58465
|
3 |
+
size 1000
|
image_0.png
ADDED
image_1.png
ADDED
image_2.png
ADDED
image_3.png
ADDED
logs/text2image-fine-tune/1696536308.1885316/events.out.tfevents.1696536308.092857a7bc4b.15301.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:859a3e5243efdb1335a9af54554d8e9c7c480e5bc97fb914c373d346f8be6c94
|
3 |
+
size 2259
|
logs/text2image-fine-tune/1696536308.190128/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: text
|
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: MaxReynolds/Lee_Souder_RocketLauncher
|
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: 15000
|
28 |
+
mixed_precision: null
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 2143
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: SouderRocketLauncherNet
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-2
|
35 |
+
push_to_hub: true
|
36 |
+
random_flip: true
|
37 |
+
rank: 4
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 256
|
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 rocket launcher by Lee Souder.
|
logs/text2image-fine-tune/1696545071.745234/events.out.tfevents.1696545071.092857a7bc4b.109216.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e4d0da1451a6e23d41e3d9b4a1cbc39a343b8ec02e5bafddae3d4a7171239c8a
|
3 |
+
size 2274
|
logs/text2image-fine-tune/1696545071.74736/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: true
|
9 |
+
checkpointing_steps: 100000
|
10 |
+
checkpoints_total_limit: null
|
11 |
+
dataloader_num_workers: 0
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: MaxReynolds/Lee_Souder_RocketLauncher
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 4
|
16 |
+
gradient_checkpointing: true
|
17 |
+
hub_model_id: null
|
18 |
+
hub_token: null
|
19 |
+
image_column: image
|
20 |
+
input_perturbation: 0
|
21 |
+
learning_rate: 1.0e-05
|
22 |
+
local_rank: -1
|
23 |
+
logging_dir: logs
|
24 |
+
lr_scheduler: constant
|
25 |
+
lr_warmup_steps: 0
|
26 |
+
max_grad_norm: 1.0
|
27 |
+
max_train_samples: null
|
28 |
+
max_train_steps: 400
|
29 |
+
mixed_precision: fp16
|
30 |
+
noise_offset: 0
|
31 |
+
non_ema_revision: null
|
32 |
+
num_train_epochs: 58
|
33 |
+
output_dir: SouderRocketLauncherNet
|
34 |
+
prediction_type: null
|
35 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-2
|
36 |
+
push_to_hub: true
|
37 |
+
random_flip: true
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 256
|
40 |
+
resume_from_checkpoint: null
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: null
|
44 |
+
snr_gamma: null
|
45 |
+
tracker_project_name: text2image-fine-tune
|
46 |
+
train_batch_size: 1
|
47 |
+
train_data_dir: null
|
48 |
+
use_8bit_adam: false
|
49 |
+
use_ema: true
|
50 |
+
validation_epochs: 5
|
logs/text2image-fine-tune/events.out.tfevents.1696536308.092857a7bc4b.15301.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b3928316876e0a2fa21b5cbbfe0c8cdd7f03713557d0b01b81cbac1be5d460e9
|
3 |
+
size 883622636
|
logs/text2image-fine-tune/events.out.tfevents.1696545071.092857a7bc4b.109216.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d877cbba4f5c78dbdbde5e5fa6ea668b205dae7054e90c3361fca475450ec40e
|
3 |
+
size 664
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fb3b8b615a44c4e1704dc99ff072bb039980ce1123d981d695c1c813d3392fd5
|
3 |
+
size 3227336
|