End of training

.gitattributes

@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+images_0.png filter=lfs diff=lfs merge=lfs -text
+images_1.png filter=lfs diff=lfs merge=lfs -text

10000/config.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "FlaxControlNetModel",
+  "_diffusers_version": "0.16.0.dev0",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "conditioning_embedding_out_channels": [
+    16,
+    32,
+    96,
+    256
+  ],
+  "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "only_cross_attention": false,
+  "sample_size": 32,
+  "use_linear_projection": false
+}

10000/diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:969ecc627b3d1b958d12e53c73c9ecfc29e5153dce10606e0ff79e673fcf2848
+size 1445128798

15000/config.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "FlaxControlNetModel",
+  "_diffusers_version": "0.16.0.dev0",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "conditioning_embedding_out_channels": [
+    16,
+    32,
+    96,
+    256
+  ],
+  "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "only_cross_attention": false,
+  "sample_size": 32,
+  "use_linear_projection": false
+}

15000/diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1267885934e8b66c6cab1582edf7289f63e041958996d92573829d3ced9cc9cc
+size 1445128798

20000/config.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "FlaxControlNetModel",
+  "_diffusers_version": "0.16.0.dev0",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "conditioning_embedding_out_channels": [
+    16,
+    32,
+    96,
+    256
+  ],
+  "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "only_cross_attention": false,
+  "sample_size": 32,
+  "use_linear_projection": false
+}

20000/diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:909711a71e3e0e95eea47c0c978939503e4a9a43108cf2d6c0331244a0a00b1b
+size 1445128798

5000/config.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "FlaxControlNetModel",
+  "_diffusers_version": "0.16.0.dev0",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "conditioning_embedding_out_channels": [
+    16,
+    32,
+    96,
+    256
+  ],
+  "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "only_cross_attention": false,
+  "sample_size": 32,
+  "use_linear_projection": false
+}

5000/diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:96638c3f3b94775ef43e6af56619703a5f5c58e643e0bcfc9c59caabd945b0b0
+size 1445128798

README.md

@@ -0,0 +1,23 @@
+
+---
+license: creativeml-openrail-m
+base_model: runwayml/stable-diffusion-v1-5
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- controlnet
+- jax-diffusers-event
+inference: true
+---
+    
+# controlnet- mfidabel/controlnet-segment-anything
+
+These are controlnet weights trained on runwayml/stable-diffusion-v1-5 with new type of conditioning. You can find some example images in the following. 
+
+prompt: modern main room of a house
+![images_0)](./images_0.png)
+prompt: a street full of people waiting for a car to pass
+![images_1)](./images_1.png)
+

config.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "FlaxControlNetModel",
+  "_diffusers_version": "0.16.0.dev0",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "conditioning_embedding_out_channels": [
+    16,
+    32,
+    96,
+    256
+  ],
+  "controlnet_conditioning_channel_order": "rgb",
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "only_cross_attention": false,
+  "sample_size": 32,
+  "use_linear_projection": false
+}

diffusion_flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6cbf91704d084ae6e6d043d9c952774129fa704b5f19dc671f0d2513e30cb07d
+size 1445128798