iamkaikai commited on
Commit
c3ed836
1 Parent(s): e657a06

End of training

Browse files
Files changed (34) hide show
  1. README.md +1 -1
  2. checkpoint-6720000/optimizer.bin +3 -0
  3. checkpoint-6720000/random_states_0.pkl +3 -0
  4. checkpoint-6720000/scaler.pt +3 -0
  5. checkpoint-6720000/scheduler.bin +3 -0
  6. checkpoint-6720000/unet/config.json +67 -0
  7. checkpoint-6720000/unet/diffusion_pytorch_model.safetensors +3 -0
  8. checkpoint-6840000/optimizer.bin +3 -0
  9. checkpoint-6840000/random_states_0.pkl +3 -0
  10. checkpoint-6840000/scaler.pt +3 -0
  11. checkpoint-6840000/scheduler.bin +3 -0
  12. checkpoint-6840000/unet/config.json +67 -0
  13. checkpoint-6840000/unet/diffusion_pytorch_model.safetensors +3 -0
  14. checkpoint-6960000/optimizer.bin +3 -0
  15. checkpoint-6960000/random_states_0.pkl +3 -0
  16. checkpoint-6960000/scaler.pt +3 -0
  17. checkpoint-6960000/scheduler.bin +3 -0
  18. checkpoint-6960000/unet/config.json +67 -0
  19. checkpoint-6960000/unet/diffusion_pytorch_model.safetensors +3 -0
  20. checkpoint-7080000/optimizer.bin +3 -0
  21. checkpoint-7080000/random_states_0.pkl +3 -0
  22. checkpoint-7080000/scaler.pt +3 -0
  23. checkpoint-7080000/scheduler.bin +3 -0
  24. checkpoint-7080000/unet/config.json +67 -0
  25. checkpoint-7080000/unet/diffusion_pytorch_model.safetensors +3 -0
  26. logs/text2image-fine-tune/1701353704.7017775/events.out.tfevents.1701353704.DESKTOP-TNN0PMP.8020.1 +3 -0
  27. logs/text2image-fine-tune/1701353704.7047784/hparams.yml +50 -0
  28. logs/text2image-fine-tune/1701353782.9124153/events.out.tfevents.1701353782.DESKTOP-TNN0PMP.5692.1 +3 -0
  29. logs/text2image-fine-tune/1701353782.9144168/hparams.yml +50 -0
  30. logs/text2image-fine-tune/events.out.tfevents.1701353704.DESKTOP-TNN0PMP.8020.0 +3 -0
  31. logs/text2image-fine-tune/events.out.tfevents.1701353782.DESKTOP-TNN0PMP.5692.0 +3 -0
  32. unet/config.json +1 -1
  33. unet/diffusion_pytorch_model.safetensors +1 -1
  34. val_imgs_grid.png +2 -2
README.md CHANGED
@@ -37,7 +37,7 @@ image.save("my_image.png")
37
 
38
  These are the key hyperparameters used during training:
39
 
40
- * Epochs: 17
41
  * Learning rate: 1e-08
42
  * Batch size: 1
43
  * Gradient accumulation steps: 1
 
37
 
38
  These are the key hyperparameters used during training:
39
 
40
+ * Epochs: 18
41
  * Learning rate: 1e-08
42
  * Batch size: 1
43
  * Gradient accumulation steps: 1
checkpoint-6720000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa42e6a5e37492f4c52f974784e3ce523c2228e27dafc65161c7334e98105b98
3
+ size 6876750164
checkpoint-6720000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a678f375318f7fc0c29590a7b049bf0fb08b4b8ac4ce633c9020c2f9d9f82cc
3
+ size 14408
checkpoint-6720000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b49f949b69f41f6c0fb3824753e3d3fcf36326b2e9c470db6bf8d15f1bf5ac36
3
+ size 988
checkpoint-6720000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0798caf224daa34389ee3319642f37c684f7bf8531a596e27a1071eebd58d974
3
+ size 1000
checkpoint-6720000/unet/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "amazing-logos-v5\\checkpoint-6700000",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "sample_size": 64,
53
+ "time_cond_proj_dim": null,
54
+ "time_embedding_act_fn": null,
55
+ "time_embedding_dim": null,
56
+ "time_embedding_type": "positional",
57
+ "timestep_post_act": null,
58
+ "transformer_layers_per_block": 1,
59
+ "up_block_types": [
60
+ "UpBlock2D",
61
+ "CrossAttnUpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D"
64
+ ],
65
+ "upcast_attention": false,
66
+ "use_linear_projection": false
67
+ }
checkpoint-6720000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ccb6e375e8d75181178219831a2e53dbd6e0a17f5ec50f6ff6601e34a7ef57d
3
+ size 3438167536
checkpoint-6840000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39d192507c3e93daa7a5327d3a47073d36edcb054effcf9a554c8525b57d75e2
3
+ size 6876750164
checkpoint-6840000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b928dd73e77cba67b0af4cbf72cbfeef4717ca23e7b311ba89c1851b7a33f5c
3
+ size 14408
checkpoint-6840000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d648bd4a0691a0eeb329b3ff2d06618281f5f89a547c2d4d5dceb5efecab7274
3
+ size 988
checkpoint-6840000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eca681e0b01b4c9324c06e7d9e49c62187b910d71e445e25139621350d21408a
3
+ size 1000
checkpoint-6840000/unet/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "amazing-logos-v5\\checkpoint-6700000",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "sample_size": 64,
53
+ "time_cond_proj_dim": null,
54
+ "time_embedding_act_fn": null,
55
+ "time_embedding_dim": null,
56
+ "time_embedding_type": "positional",
57
+ "timestep_post_act": null,
58
+ "transformer_layers_per_block": 1,
59
+ "up_block_types": [
60
+ "UpBlock2D",
61
+ "CrossAttnUpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D"
64
+ ],
65
+ "upcast_attention": false,
66
+ "use_linear_projection": false
67
+ }
checkpoint-6840000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df3009b7d45c113e84c143cffd49a7327e748516a20f65be5eaf38c62f862f8c
3
+ size 3438167536
checkpoint-6960000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0328f8ee2fdc8e7f24458cb80213affa0e9eaa09080e5d3500512fc64cbc994f
3
+ size 6876750164
checkpoint-6960000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de872034052ce25cd169130ef1b466ae0c1c0b35fa4eee7db8d27c5d3054e069
3
+ size 14408
checkpoint-6960000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0cd8181e37e51e5578df0eef8dccc1640048185d77650ed1d5ff76a6f4d8429
3
+ size 988
checkpoint-6960000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb952384915442d5f51bdf35c177e55fb1707ceb245e530ab8e8c19c4571dd9e
3
+ size 1000
checkpoint-6960000/unet/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "amazing-logos-v5\\checkpoint-6700000",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "sample_size": 64,
53
+ "time_cond_proj_dim": null,
54
+ "time_embedding_act_fn": null,
55
+ "time_embedding_dim": null,
56
+ "time_embedding_type": "positional",
57
+ "timestep_post_act": null,
58
+ "transformer_layers_per_block": 1,
59
+ "up_block_types": [
60
+ "UpBlock2D",
61
+ "CrossAttnUpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D"
64
+ ],
65
+ "upcast_attention": false,
66
+ "use_linear_projection": false
67
+ }
checkpoint-6960000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d88834a138423c9d4e36eff4a34fa5bb0e82c0d8d82b27e09aedd9f4ca01ff02
3
+ size 3438167536
checkpoint-7080000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4212f767439d16b69b80302bce4c449d16582a158f7dd4faafd4d0da0d9f7a2b
3
+ size 6876750164
checkpoint-7080000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57b83e67b9fc7de2cd64ebe9cecb29d2748c6a6cec45f258ddd04ae2505ffd12
3
+ size 14408
checkpoint-7080000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ef7f9ccea6e2650db10d740fc3a6fd6432f9aaf6b09f4b9334d2c3ff59140c7
3
+ size 988
checkpoint-7080000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2227b815d7c46afcdf32d01dc4ebc7acd4dc93182daf3571cadc6a1815535f2c
3
+ size 1000
checkpoint-7080000/unet/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "amazing-logos-v5\\checkpoint-6700000",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "sample_size": 64,
53
+ "time_cond_proj_dim": null,
54
+ "time_embedding_act_fn": null,
55
+ "time_embedding_dim": null,
56
+ "time_embedding_type": "positional",
57
+ "timestep_post_act": null,
58
+ "transformer_layers_per_block": 1,
59
+ "up_block_types": [
60
+ "UpBlock2D",
61
+ "CrossAttnUpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D"
64
+ ],
65
+ "upcast_attention": false,
66
+ "use_linear_projection": false
67
+ }
checkpoint-7080000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea646d06b2ebe397f72b64624fd84606f6967e9951aa1970a4bc4ed7b59f5d0d
3
+ size 3438167536
logs/text2image-fine-tune/1701353704.7017775/events.out.tfevents.1701353704.DESKTOP-TNN0PMP.8020.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e40d55a837aeb5e1bfbe28725faa28535f200fc24aada1b6dfb3e44c6b40cbfe
3
+ size 2416
logs/text2image-fine-tune/1701353704.7047784/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: true
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: iamkaikai/amazing_logos_v4
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: amazing-logos-v5
18
+ hub_token: null
19
+ image_column: image
20
+ input_perturbation: 0
21
+ learning_rate: 1.0e-08
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 400000
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 6753267
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 17
33
+ output_dir: amazing-logos-v5
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: false
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: checkpoint-6700000
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 6666
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: false
50
+ validation_epochs: 1
logs/text2image-fine-tune/1701353782.9124153/events.out.tfevents.1701353782.DESKTOP-TNN0PMP.5692.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac1bde1a830590f205626a935a6d74ff8cfb3fc97df381fa100f1928ab20c2f4
3
+ size 2416
logs/text2image-fine-tune/1701353782.9144168/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: true
6
+ cache_dir: null
7
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 120000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: iamkaikai/amazing_logos_v4
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: amazing-logos-v5
18
+ hub_token: null
19
+ image_column: image
20
+ input_perturbation: 0
21
+ learning_rate: 1.0e-08
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 400000
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 7150518
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 18
33
+ output_dir: amazing-logos-v5
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: false
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: checkpoint-6700000
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 6666
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: false
50
+ validation_epochs: 1
logs/text2image-fine-tune/events.out.tfevents.1701353704.DESKTOP-TNN0PMP.8020.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23fc58ae81ed8bb0d39718f81a86738cf17fce4545b59db3a0bd114bfd6ed710
3
+ size 2434
logs/text2image-fine-tune/events.out.tfevents.1701353782.DESKTOP-TNN0PMP.5692.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4de7517348e5adf106611b7bb0d149fed25234322874ab2312255eecbc408330
3
+ size 36789150
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.22.0.dev0",
4
- "_name_or_path": "amazing-logos-v5\\checkpoint-6320000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "amazing-logos-v5\\checkpoint-6700000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f179668f8470c4e409ded2808181904c79c26820ef5bf48f78c2051510f7850
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6adbd705bb79927b6040931fb40b8ed5b146044495edd19022de7a2bbdd41a34
3
  size 3438167536
val_imgs_grid.png CHANGED

Git LFS Details

  • SHA256: fb4ee7876e3f043981b13c2eb4297e760bfa63ce19b2a387aa9afdcd5b2c452d
  • Pointer size: 132 Bytes
  • Size of remote file: 6.89 MB

Git LFS Details

  • SHA256: b43fcd13951d613d1bfe9743343467cc98178389f0e3e2636bbcb63881cf3ad1
  • Pointer size: 132 Bytes
  • Size of remote file: 6.97 MB