moritzef commited on
Commit
16c50b9
1 Parent(s): fd36949

End of training

Browse files
README.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ library_name: diffusers
4
+ tags:
5
+ - stable-diffusion
6
+ - stable-diffusion-diffusers
7
+ - text-to-image
8
+ - diffusers
9
+ - controlnet
10
+ - diffusers-training
11
+ base_model: stabilityai/stable-diffusion-2-1-base
12
+ inference: true
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+
19
+ # controlnet-moritzef/model_lr1e4
20
+
21
+ These are controlnet weights trained on stabilityai/stable-diffusion-2-1-base with new type of conditioning.
22
+ You can find some example images below.
23
+
24
+ prompt: A realistic google streetview image taken in Berlin (Germany), that looks normal and has a beauty-score of 24, where scores are between 10 and 40 and higher scores indicate more beauty.
25
+ ![images_0)](./images_0.png)
26
+ prompt: A realistic google streetview image taken in New York (USA), that looks normal and has a beauty-score of 27, where scores are between 10 and 40 and higher scores indicate more beauty.
27
+ ![images_1)](./images_1.png)
28
+ prompt: A realistic google streetview image taken in Rome (Italy), that looks normal and has a beauty-score of 23, where scores are between 10 and 40 and higher scores indicate more beauty.
29
+ ![images_2)](./images_2.png)
30
+ prompt: A realistic google streetview image taken in Mexico City (Mexico), that looks very ugly and has a beauty-score of 18, where scores are between 10 and 40 and higher scores indicate more beauty.
31
+ ![images_3)](./images_3.png)
32
+ prompt: A realistic google streetview image taken in Tel Aviv (Israel), that looks normal and has a beauty-score of 24, where scores are between 10 and 40 and higher scores indicate more beauty.
33
+ ![images_4)](./images_4.png)
34
+ prompt: A realistic google streetview image taken in Kyoto (Japan), that looks very ugly and has a beauty-score of 16, where scores are between 10 and 40 and higher scores indicate more beauty.
35
+ ![images_5)](./images_5.png)
36
+ prompt: A realistic google streetview image taken in Gaborone (Botswana), that looks normal and has a beauty-score of 24, where scores are between 10 and 40 and higher scores indicate more beauty.
37
+ ![images_6)](./images_6.png)
38
+ prompt: A realistic google streetview image taken in Melbourne (Australia), that looks normal and has a beauty-score of 28, where scores are between 10 and 40 and higher scores indicate more beauty.
39
+ ![images_7)](./images_7.png)
40
+
41
+
42
+
43
+ ## Intended uses & limitations
44
+
45
+ #### How to use
46
+
47
+ ```python
48
+ # TODO: add an example code snippet for running this diffusion pipeline
49
+ ```
50
+
51
+ #### Limitations and bias
52
+
53
+ [TODO: provide examples of latent issues and potential remediations]
54
+
55
+ ## Training details
56
+
57
+ [TODO: describe the data used to train the model]
checkpoint-100000/controlnet/config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "addition_time_embed_dim": null,
8
+ "attention_head_dim": [
9
+ 5,
10
+ 10,
11
+ 20,
12
+ 20
13
+ ],
14
+ "block_out_channels": [
15
+ 320,
16
+ 640,
17
+ 1280,
18
+ 1280
19
+ ],
20
+ "class_embed_type": null,
21
+ "conditioning_channels": 3,
22
+ "conditioning_embedding_out_channels": [
23
+ 16,
24
+ 32,
25
+ 96,
26
+ 256
27
+ ],
28
+ "controlnet_conditioning_channel_order": "rgb",
29
+ "cross_attention_dim": 1024,
30
+ "down_block_types": [
31
+ "CrossAttnDownBlock2D",
32
+ "CrossAttnDownBlock2D",
33
+ "CrossAttnDownBlock2D",
34
+ "DownBlock2D"
35
+ ],
36
+ "downsample_padding": 1,
37
+ "encoder_hid_dim": null,
38
+ "encoder_hid_dim_type": null,
39
+ "flip_sin_to_cos": true,
40
+ "freq_shift": 0,
41
+ "global_pool_conditions": false,
42
+ "in_channels": 4,
43
+ "layers_per_block": 2,
44
+ "mid_block_scale_factor": 1,
45
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
46
+ "norm_eps": 1e-05,
47
+ "norm_num_groups": 32,
48
+ "num_attention_heads": null,
49
+ "num_class_embeds": null,
50
+ "only_cross_attention": false,
51
+ "projection_class_embeddings_input_dim": null,
52
+ "resnet_time_scale_shift": "default",
53
+ "transformer_layers_per_block": 1,
54
+ "upcast_attention": false,
55
+ "use_linear_projection": true
56
+ }
checkpoint-100000/controlnet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43c49e7fae99c39280b6e66695868470bba81794aa096118310a233e6c06267f
3
+ size 1456953560
checkpoint-100000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf1bbd2e414e51ed45ff29716247c6abbe1a7c475e6066da285c4d46ef06d56a
3
+ size 2914111182
checkpoint-100000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a25e10b472d9b197f5f1c83d53a132ddf247b71c55eb71adf3f7b590771654a
3
+ size 14408
checkpoint-100000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b35b67e2e1b2d26fc187ae57fff5b681d4784f24f7821ee9f2eb1e824cca739
3
+ size 1000
config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "addition_time_embed_dim": null,
8
+ "attention_head_dim": [
9
+ 5,
10
+ 10,
11
+ 20,
12
+ 20
13
+ ],
14
+ "block_out_channels": [
15
+ 320,
16
+ 640,
17
+ 1280,
18
+ 1280
19
+ ],
20
+ "class_embed_type": null,
21
+ "conditioning_channels": 3,
22
+ "conditioning_embedding_out_channels": [
23
+ 16,
24
+ 32,
25
+ 96,
26
+ 256
27
+ ],
28
+ "controlnet_conditioning_channel_order": "rgb",
29
+ "cross_attention_dim": 1024,
30
+ "down_block_types": [
31
+ "CrossAttnDownBlock2D",
32
+ "CrossAttnDownBlock2D",
33
+ "CrossAttnDownBlock2D",
34
+ "DownBlock2D"
35
+ ],
36
+ "downsample_padding": 1,
37
+ "encoder_hid_dim": null,
38
+ "encoder_hid_dim_type": null,
39
+ "flip_sin_to_cos": true,
40
+ "freq_shift": 0,
41
+ "global_pool_conditions": false,
42
+ "in_channels": 4,
43
+ "layers_per_block": 2,
44
+ "mid_block_scale_factor": 1,
45
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
46
+ "norm_eps": 1e-05,
47
+ "norm_num_groups": 32,
48
+ "num_attention_heads": null,
49
+ "num_class_embeds": null,
50
+ "only_cross_attention": false,
51
+ "projection_class_embeddings_input_dim": null,
52
+ "resnet_time_scale_shift": "default",
53
+ "transformer_layers_per_block": 1,
54
+ "upcast_attention": false,
55
+ "use_linear_projection": true
56
+ }
diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db1e19a206f40b021ddae15a7e30068ce96634e520c232f2f3481bb45ce984b5
3
+ size 1456953560
image_control.png ADDED
images_0.png ADDED
images_1.png ADDED
images_2.png ADDED
images_3.png ADDED
images_4.png ADDED
images_5.png ADDED
images_6.png ADDED
images_7.png ADDED