MaxReynolds commited on
Commit
49187ff
1 Parent(s): d73e210

End of training

Browse files
Files changed (42) hide show
  1. README.md +21 -0
  2. checkpoint-1000/optimizer.bin +3 -0
  3. checkpoint-1000/pytorch_model.bin +3 -0
  4. checkpoint-1000/random_states_0.pkl +3 -0
  5. checkpoint-1000/scaler.pt +3 -0
  6. checkpoint-1000/scheduler.bin +3 -0
  7. checkpoint-1500/optimizer.bin +3 -0
  8. checkpoint-1500/pytorch_model.bin +3 -0
  9. checkpoint-1500/random_states_0.pkl +3 -0
  10. checkpoint-1500/scaler.pt +3 -0
  11. checkpoint-1500/scheduler.bin +3 -0
  12. checkpoint-2000/optimizer.bin +3 -0
  13. checkpoint-2000/pytorch_model.bin +3 -0
  14. checkpoint-2000/random_states_0.pkl +3 -0
  15. checkpoint-2000/scaler.pt +3 -0
  16. checkpoint-2000/scheduler.bin +3 -0
  17. checkpoint-2500/optimizer.bin +3 -0
  18. checkpoint-2500/pytorch_model.bin +3 -0
  19. checkpoint-2500/random_states_0.pkl +3 -0
  20. checkpoint-2500/scaler.pt +3 -0
  21. checkpoint-2500/scheduler.bin +3 -0
  22. checkpoint-3000/optimizer.bin +3 -0
  23. checkpoint-3000/pytorch_model.bin +3 -0
  24. checkpoint-3000/random_states_0.pkl +3 -0
  25. checkpoint-3000/scaler.pt +3 -0
  26. checkpoint-3000/scheduler.bin +3 -0
  27. checkpoint-500/optimizer.bin +3 -0
  28. checkpoint-500/pytorch_model.bin +3 -0
  29. checkpoint-500/random_states_0.pkl +3 -0
  30. checkpoint-500/scaler.pt +3 -0
  31. checkpoint-500/scheduler.bin +3 -0
  32. image_0.png +0 -0
  33. image_1.png +0 -0
  34. image_2.png +0 -0
  35. image_3.png +0 -0
  36. logs/text2image-fine-tune/1696536308.1885316/events.out.tfevents.1696536308.092857a7bc4b.15301.1 +3 -0
  37. logs/text2image-fine-tune/1696536308.190128/hparams.yml +49 -0
  38. logs/text2image-fine-tune/1696545071.745234/events.out.tfevents.1696545071.092857a7bc4b.109216.1 +3 -0
  39. logs/text2image-fine-tune/1696545071.74736/hparams.yml +50 -0
  40. logs/text2image-fine-tune/events.out.tfevents.1696536308.092857a7bc4b.15301.0 +3 -0
  41. logs/text2image-fine-tune/events.out.tfevents.1696545071.092857a7bc4b.109216.0 +3 -0
  42. 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