End of training
Browse files- README.md +4 -4
- checkpoint-130000/optimizer.bin +3 -0
- checkpoint-130000/random_states_0.pkl +3 -0
- checkpoint-130000/scaler.pt +3 -0
- checkpoint-130000/scheduler.bin +3 -0
- checkpoint-130000/unet/config.json +65 -0
- checkpoint-130000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-140000/optimizer.bin +3 -0
- checkpoint-140000/random_states_0.pkl +3 -0
- checkpoint-140000/scaler.pt +3 -0
- checkpoint-140000/scheduler.bin +3 -0
- checkpoint-140000/unet/config.json +65 -0
- checkpoint-140000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-150000/optimizer.bin +3 -0
- checkpoint-150000/random_states_0.pkl +3 -0
- checkpoint-150000/scaler.pt +3 -0
- checkpoint-150000/scheduler.bin +3 -0
- checkpoint-150000/unet/config.json +65 -0
- checkpoint-150000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-160000/optimizer.bin +3 -0
- checkpoint-160000/random_states_0.pkl +3 -0
- checkpoint-160000/scaler.pt +3 -0
- checkpoint-160000/scheduler.bin +3 -0
- checkpoint-160000/unet/config.json +65 -0
- checkpoint-160000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-170000/optimizer.bin +3 -0
- checkpoint-170000/random_states_0.pkl +3 -0
- checkpoint-170000/scaler.pt +3 -0
- checkpoint-170000/scheduler.bin +3 -0
- checkpoint-170000/unet/config.json +65 -0
- checkpoint-170000/unet/diffusion_pytorch_model.bin +3 -0
- unet/config.json +1 -1
- unet/diffusion_pytorch_model.bin +1 -1
- val_imgs_grid.png +0 -0
README.md
CHANGED
@@ -14,7 +14,7 @@ inference: true
|
|
14 |
|
15 |
# Text-to-image finetuning - iamkaikai/amazing-logos
|
16 |
|
17 |
-
This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **iamkaikai/amazing_logos_v2** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['Simple elegant logo for Digital Art, D A
|
18 |
|
19 |
![val_imgs_grid](./val_imgs_grid.png)
|
20 |
|
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
|
|
28 |
import torch
|
29 |
|
30 |
pipeline = DiffusionPipeline.from_pretrained("iamkaikai/amazing-logos", torch_dtype=torch.float16)
|
31 |
-
prompt = "Simple elegant logo for Digital Art, D A
|
32 |
image = pipeline(prompt).images[0]
|
33 |
image.save("my_image.png")
|
34 |
```
|
@@ -37,7 +37,7 @@ image.save("my_image.png")
|
|
37 |
|
38 |
These are the key hyperparameters used during training:
|
39 |
|
40 |
-
* Epochs:
|
41 |
* Learning rate: 1e-07
|
42 |
* Batch size: 1
|
43 |
* Gradient accumulation steps: 1
|
@@ -45,4 +45,4 @@ These are the key hyperparameters used during training:
|
|
45 |
* Mixed-precision: fp16
|
46 |
|
47 |
|
48 |
-
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/iam-kai-kai/text2image-fine-tune/runs/
|
|
|
14 |
|
15 |
# Text-to-image finetuning - iamkaikai/amazing-logos
|
16 |
|
17 |
+
This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **iamkaikai/amazing_logos_v2** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['Simple elegant logo for Digital Art, D A triangle Symmetrical, successful vibe, minimalist, thought-provoking, abstract, recognizable, black and white']:
|
18 |
|
19 |
![val_imgs_grid](./val_imgs_grid.png)
|
20 |
|
|
|
28 |
import torch
|
29 |
|
30 |
pipeline = DiffusionPipeline.from_pretrained("iamkaikai/amazing-logos", torch_dtype=torch.float16)
|
31 |
+
prompt = "Simple elegant logo for Digital Art, D A triangle Symmetrical, successful vibe, minimalist, thought-provoking, abstract, recognizable, black and white"
|
32 |
image = pipeline(prompt).images[0]
|
33 |
image.save("my_image.png")
|
34 |
```
|
|
|
37 |
|
38 |
These are the key hyperparameters used during training:
|
39 |
|
40 |
+
* Epochs: 17
|
41 |
* Learning rate: 1e-07
|
42 |
* Batch size: 1
|
43 |
* Gradient accumulation steps: 1
|
|
|
45 |
* Mixed-precision: fp16
|
46 |
|
47 |
|
48 |
+
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/iam-kai-kai/text2image-fine-tune/runs/o936ikb3).
|
checkpoint-130000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e9c8da66ebd32d8ab9cfce568f3caedf48e171a2aae154a8cb9d76bfca2f25df
|
3 |
+
size 6876749715
|
checkpoint-130000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:02dbbae7ab1e5c2eb00f5d8c6845c4510a055447f2a999b3068612e8422b7d75
|
3 |
+
size 14727
|
checkpoint-130000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2322ac0c9c403579a6f2166859c4a2cd54c9204e891f62092e9b99a164f3a968
|
3 |
+
size 557
|
checkpoint-130000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c486bf5ca564fc71caac193897a471d2369b22cc9f734e8f7b7eef57d9f5674f
|
3 |
+
size 563
|
checkpoint-130000/unet/config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
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 |
+
"block_out_channels": [
|
11 |
+
320,
|
12 |
+
640,
|
13 |
+
1280,
|
14 |
+
1280
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"class_embeddings_concat": false,
|
19 |
+
"conv_in_kernel": 3,
|
20 |
+
"conv_out_kernel": 3,
|
21 |
+
"cross_attention_dim": 768,
|
22 |
+
"cross_attention_norm": null,
|
23 |
+
"down_block_types": [
|
24 |
+
"CrossAttnDownBlock2D",
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"DownBlock2D"
|
28 |
+
],
|
29 |
+
"downsample_padding": 1,
|
30 |
+
"dual_cross_attention": false,
|
31 |
+
"encoder_hid_dim": null,
|
32 |
+
"encoder_hid_dim_type": null,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"layers_per_block": 2,
|
37 |
+
"mid_block_only_cross_attention": null,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"norm_eps": 1e-05,
|
41 |
+
"norm_num_groups": 32,
|
42 |
+
"num_attention_heads": null,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"out_channels": 4,
|
46 |
+
"projection_class_embeddings_input_dim": null,
|
47 |
+
"resnet_out_scale_factor": 1.0,
|
48 |
+
"resnet_skip_time_act": false,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_act_fn": null,
|
53 |
+
"time_embedding_dim": null,
|
54 |
+
"time_embedding_type": "positional",
|
55 |
+
"timestep_post_act": null,
|
56 |
+
"transformer_layers_per_block": 1,
|
57 |
+
"up_block_types": [
|
58 |
+
"UpBlock2D",
|
59 |
+
"CrossAttnUpBlock2D",
|
60 |
+
"CrossAttnUpBlock2D",
|
61 |
+
"CrossAttnUpBlock2D"
|
62 |
+
],
|
63 |
+
"upcast_attention": false,
|
64 |
+
"use_linear_projection": false
|
65 |
+
}
|
checkpoint-130000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5c2695c3ac58645580af360e826165e45c589e63f35834a71fca98af8f2ad8bd
|
3 |
+
size 3438375973
|
checkpoint-140000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebc3d6983ccd3a88862e96a717ee5069b4cd074f5ce5c95bbaa0bb890170d77b
|
3 |
+
size 6876749715
|
checkpoint-140000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b016563b7f885dba8bfdb95465ed4f84268a16d325b5941d87f3132c5cf03d9a
|
3 |
+
size 14727
|
checkpoint-140000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ff16f92ec2e5f78d2da736612e48abbfd5ddf666e862c541316eb2b4c593a073
|
3 |
+
size 557
|
checkpoint-140000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f56c7331dc6e0bccc1b382e8ad46169ad9adbc51dab59a11439993987c8108a9
|
3 |
+
size 563
|
checkpoint-140000/unet/config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
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 |
+
"block_out_channels": [
|
11 |
+
320,
|
12 |
+
640,
|
13 |
+
1280,
|
14 |
+
1280
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"class_embeddings_concat": false,
|
19 |
+
"conv_in_kernel": 3,
|
20 |
+
"conv_out_kernel": 3,
|
21 |
+
"cross_attention_dim": 768,
|
22 |
+
"cross_attention_norm": null,
|
23 |
+
"down_block_types": [
|
24 |
+
"CrossAttnDownBlock2D",
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"DownBlock2D"
|
28 |
+
],
|
29 |
+
"downsample_padding": 1,
|
30 |
+
"dual_cross_attention": false,
|
31 |
+
"encoder_hid_dim": null,
|
32 |
+
"encoder_hid_dim_type": null,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"layers_per_block": 2,
|
37 |
+
"mid_block_only_cross_attention": null,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"norm_eps": 1e-05,
|
41 |
+
"norm_num_groups": 32,
|
42 |
+
"num_attention_heads": null,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"out_channels": 4,
|
46 |
+
"projection_class_embeddings_input_dim": null,
|
47 |
+
"resnet_out_scale_factor": 1.0,
|
48 |
+
"resnet_skip_time_act": false,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_act_fn": null,
|
53 |
+
"time_embedding_dim": null,
|
54 |
+
"time_embedding_type": "positional",
|
55 |
+
"timestep_post_act": null,
|
56 |
+
"transformer_layers_per_block": 1,
|
57 |
+
"up_block_types": [
|
58 |
+
"UpBlock2D",
|
59 |
+
"CrossAttnUpBlock2D",
|
60 |
+
"CrossAttnUpBlock2D",
|
61 |
+
"CrossAttnUpBlock2D"
|
62 |
+
],
|
63 |
+
"upcast_attention": false,
|
64 |
+
"use_linear_projection": false
|
65 |
+
}
|
checkpoint-140000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2f559cd4b82e86487254a4f4b2f30d5a9c6d3c50bd001e00b574e4fcf164721
|
3 |
+
size 3438375973
|
checkpoint-150000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:744b4c4ad7b2502ff2ca02a67ca2713a39369e3387b1e7d479f8de31bffc1f53
|
3 |
+
size 6876749715
|
checkpoint-150000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:571ae8f17e867261502e21c7d9962e851a8ad7c2192f485d682c5757bf395e76
|
3 |
+
size 14727
|
checkpoint-150000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:613fa8d15b5598fbfe30b7d1aeefbca3b3fc9f85fa3d60161f85481d303d5c0e
|
3 |
+
size 557
|
checkpoint-150000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:843431f732b4c156724b99c7fc65615b8a7417d8e81ee8b313cf06669c7e711d
|
3 |
+
size 563
|
checkpoint-150000/unet/config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
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 |
+
"block_out_channels": [
|
11 |
+
320,
|
12 |
+
640,
|
13 |
+
1280,
|
14 |
+
1280
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"class_embeddings_concat": false,
|
19 |
+
"conv_in_kernel": 3,
|
20 |
+
"conv_out_kernel": 3,
|
21 |
+
"cross_attention_dim": 768,
|
22 |
+
"cross_attention_norm": null,
|
23 |
+
"down_block_types": [
|
24 |
+
"CrossAttnDownBlock2D",
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"DownBlock2D"
|
28 |
+
],
|
29 |
+
"downsample_padding": 1,
|
30 |
+
"dual_cross_attention": false,
|
31 |
+
"encoder_hid_dim": null,
|
32 |
+
"encoder_hid_dim_type": null,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"layers_per_block": 2,
|
37 |
+
"mid_block_only_cross_attention": null,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"norm_eps": 1e-05,
|
41 |
+
"norm_num_groups": 32,
|
42 |
+
"num_attention_heads": null,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"out_channels": 4,
|
46 |
+
"projection_class_embeddings_input_dim": null,
|
47 |
+
"resnet_out_scale_factor": 1.0,
|
48 |
+
"resnet_skip_time_act": false,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_act_fn": null,
|
53 |
+
"time_embedding_dim": null,
|
54 |
+
"time_embedding_type": "positional",
|
55 |
+
"timestep_post_act": null,
|
56 |
+
"transformer_layers_per_block": 1,
|
57 |
+
"up_block_types": [
|
58 |
+
"UpBlock2D",
|
59 |
+
"CrossAttnUpBlock2D",
|
60 |
+
"CrossAttnUpBlock2D",
|
61 |
+
"CrossAttnUpBlock2D"
|
62 |
+
],
|
63 |
+
"upcast_attention": false,
|
64 |
+
"use_linear_projection": false
|
65 |
+
}
|
checkpoint-150000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19de2b59c8d858f3b085647f3aff44bc012ceeae272c80d97e2c5e861773aa53
|
3 |
+
size 3438375973
|
checkpoint-160000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:708fbfd2b30f90955bbd01df6f9563aa64b1e496e547c9f2e897197d2ff1c42e
|
3 |
+
size 6876749715
|
checkpoint-160000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:20790fce6b48a8abdb54e287ef811d50fba2adbe47b60a629f7314bd2b6b37cf
|
3 |
+
size 14727
|
checkpoint-160000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8cdef0d479d1db1bc77d173fa45999a160d2e1a87754cc37e9800bafc97a3b33
|
3 |
+
size 557
|
checkpoint-160000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:07b76f559680ab4c8e9bb743ee8f24554dbd8ae12a00533884592bba5ee8ec21
|
3 |
+
size 563
|
checkpoint-160000/unet/config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
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 |
+
"block_out_channels": [
|
11 |
+
320,
|
12 |
+
640,
|
13 |
+
1280,
|
14 |
+
1280
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"class_embeddings_concat": false,
|
19 |
+
"conv_in_kernel": 3,
|
20 |
+
"conv_out_kernel": 3,
|
21 |
+
"cross_attention_dim": 768,
|
22 |
+
"cross_attention_norm": null,
|
23 |
+
"down_block_types": [
|
24 |
+
"CrossAttnDownBlock2D",
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"DownBlock2D"
|
28 |
+
],
|
29 |
+
"downsample_padding": 1,
|
30 |
+
"dual_cross_attention": false,
|
31 |
+
"encoder_hid_dim": null,
|
32 |
+
"encoder_hid_dim_type": null,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"layers_per_block": 2,
|
37 |
+
"mid_block_only_cross_attention": null,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"norm_eps": 1e-05,
|
41 |
+
"norm_num_groups": 32,
|
42 |
+
"num_attention_heads": null,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"out_channels": 4,
|
46 |
+
"projection_class_embeddings_input_dim": null,
|
47 |
+
"resnet_out_scale_factor": 1.0,
|
48 |
+
"resnet_skip_time_act": false,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_act_fn": null,
|
53 |
+
"time_embedding_dim": null,
|
54 |
+
"time_embedding_type": "positional",
|
55 |
+
"timestep_post_act": null,
|
56 |
+
"transformer_layers_per_block": 1,
|
57 |
+
"up_block_types": [
|
58 |
+
"UpBlock2D",
|
59 |
+
"CrossAttnUpBlock2D",
|
60 |
+
"CrossAttnUpBlock2D",
|
61 |
+
"CrossAttnUpBlock2D"
|
62 |
+
],
|
63 |
+
"upcast_attention": false,
|
64 |
+
"use_linear_projection": false
|
65 |
+
}
|
checkpoint-160000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a84451ad198c397bb1859324c80f47e5f5ae73c6e18190fc9327f0acaea94cc4
|
3 |
+
size 3438375973
|
checkpoint-170000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a5f62d5bd4d8a2c4188a76c933fb1f0d699a88f0988ed2ace3d6100a73919ce7
|
3 |
+
size 6876749715
|
checkpoint-170000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fd777a1a2007485ca28fb43410c940321f7a8f55157f00d2efaa28e9255a656
|
3 |
+
size 14727
|
checkpoint-170000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fd11e3c405d9f506b5b815d59e354ce6e2d48b3a335686c77affeb2f4d802f00
|
3 |
+
size 557
|
checkpoint-170000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a4308a8a3d982fe3d0f5727f35f5d4792012f7349729c0cde9c452565a2c3e21
|
3 |
+
size 563
|
checkpoint-170000/unet/config.json
ADDED
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
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 |
+
"block_out_channels": [
|
11 |
+
320,
|
12 |
+
640,
|
13 |
+
1280,
|
14 |
+
1280
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"class_embeddings_concat": false,
|
19 |
+
"conv_in_kernel": 3,
|
20 |
+
"conv_out_kernel": 3,
|
21 |
+
"cross_attention_dim": 768,
|
22 |
+
"cross_attention_norm": null,
|
23 |
+
"down_block_types": [
|
24 |
+
"CrossAttnDownBlock2D",
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"DownBlock2D"
|
28 |
+
],
|
29 |
+
"downsample_padding": 1,
|
30 |
+
"dual_cross_attention": false,
|
31 |
+
"encoder_hid_dim": null,
|
32 |
+
"encoder_hid_dim_type": null,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"layers_per_block": 2,
|
37 |
+
"mid_block_only_cross_attention": null,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"norm_eps": 1e-05,
|
41 |
+
"norm_num_groups": 32,
|
42 |
+
"num_attention_heads": null,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"out_channels": 4,
|
46 |
+
"projection_class_embeddings_input_dim": null,
|
47 |
+
"resnet_out_scale_factor": 1.0,
|
48 |
+
"resnet_skip_time_act": false,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_act_fn": null,
|
53 |
+
"time_embedding_dim": null,
|
54 |
+
"time_embedding_type": "positional",
|
55 |
+
"timestep_post_act": null,
|
56 |
+
"transformer_layers_per_block": 1,
|
57 |
+
"up_block_types": [
|
58 |
+
"UpBlock2D",
|
59 |
+
"CrossAttnUpBlock2D",
|
60 |
+
"CrossAttnUpBlock2D",
|
61 |
+
"CrossAttnUpBlock2D"
|
62 |
+
],
|
63 |
+
"upcast_attention": false,
|
64 |
+
"use_linear_projection": false
|
65 |
+
}
|
checkpoint-170000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:214f19871f3d809211f950e2b83e713c41ddf95a873eebf2b2232427d64ac580
|
3 |
+
size 3438375973
|
unet/config.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
"_diffusers_version": "0.19.0.dev0",
|
4 |
-
"_name_or_path": "sd/amazing-logos/checkpoint-
|
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.19.0.dev0",
|
4 |
+
"_name_or_path": "sd/amazing-logos/checkpoint-120000",
|
5 |
"act_fn": "silu",
|
6 |
"addition_embed_type": null,
|
7 |
"addition_embed_type_num_heads": 64,
|
unet/diffusion_pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3438375973
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:214f19871f3d809211f950e2b83e713c41ddf95a873eebf2b2232427d64ac580
|
3 |
size 3438375973
|
val_imgs_grid.png
CHANGED