diff --git a/.gitattributes b/.gitattributes
index c7d9f3332a950355d5a77d85000f05e6f45435ea..28a386126e4dc02feff2d5357fb0e31f7db4dce5 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -32,3 +32,4 @@ 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
diff --git a/10000/config.json b/10000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/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
+}
diff --git a/10000/diffusion_flax_model.msgpack b/10000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..c342a45adbb9e0a9de745a89911cd7d6e2ac24bc
--- /dev/null
+++ b/10000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:96c54af46000d1902c8588c6c7795b10624989dc0cdc0547741c997d29a849f9
+size 1445128798
diff --git a/100000/config.json b/100000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/100000/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
+}
diff --git a/100000/diffusion_flax_model.msgpack b/100000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..488e0445fc26f27bcb4b2854dca15913af85c3d1
--- /dev/null
+++ b/100000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c29f78b8631ee0031cfc2f0825638fa3abc5d1a97979b54331b7c2ea5ca4e057
+size 1445128798
diff --git a/105000/config.json b/105000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/105000/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
+}
diff --git a/105000/diffusion_flax_model.msgpack b/105000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..0af243bffc0a2bae0bd44537c7a7374141b7e458
--- /dev/null
+++ b/105000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5e66da76185fc7ba6e17fccd5c0807c4cbfd7e4a3ed5007fb2267cce1ae9d2e
+size 1445128798
diff --git a/110000/config.json b/110000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/110000/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
+}
diff --git a/110000/diffusion_flax_model.msgpack b/110000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..e0149f673307c832f38fc3e139143e21d5fbebb1
--- /dev/null
+++ b/110000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6782b5a35afb0ad58725db609e389d94a8483cacd57a41d9c4b1329e06791b59
+size 1445128798
diff --git a/115000/config.json b/115000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/115000/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
+}
diff --git a/115000/diffusion_flax_model.msgpack b/115000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..1935333599c2bb24f4c13d8ac300196d7f817fb7
--- /dev/null
+++ b/115000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf4b832e22396ed11a39986bbed9964a5874e1580c1504610e4b93272fb3b527
+size 1445128798
diff --git a/120000/config.json b/120000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/120000/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
+}
diff --git a/120000/diffusion_flax_model.msgpack b/120000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..3963d5064eac6a2e86ce5daa11ebb68de398bb90
--- /dev/null
+++ b/120000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f99a2ef4fb9da7d8564c5f5abc8b95b4a32cba1655d4e0f82cd947ef921c591
+size 1445128798
diff --git a/125000/config.json b/125000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/125000/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
+}
diff --git a/125000/diffusion_flax_model.msgpack b/125000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..378b41edcf9a0c7722ffce30ec27cb592107ab38
--- /dev/null
+++ b/125000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4767ceb6077bf05a1143b1e293072bae9192d67e73422e2ed4cc7af827cd138e
+size 1445128798
diff --git a/130000/config.json b/130000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/130000/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
+}
diff --git a/130000/diffusion_flax_model.msgpack b/130000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d84ed8bc7cac3cb1d3eb4b1a3155851d40f0c8bf
--- /dev/null
+++ b/130000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfbffe8abe033dcd30f7a83a899c4be463f9c1eed6b06d4ba511d0c0d9904048
+size 1445128798
diff --git a/135000/config.json b/135000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/135000/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
+}
diff --git a/135000/diffusion_flax_model.msgpack b/135000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..fafc9e6857b6161e0f9601da542d82ad4a2356d0
--- /dev/null
+++ b/135000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0379f33ca833c4c0de86ab3ab0a8f9e9e869d6c5cdfa8c428b2cf735adca4cef
+size 1445128798
diff --git a/140000/config.json b/140000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/140000/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
+}
diff --git a/140000/diffusion_flax_model.msgpack b/140000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..6a8b9f69d37cd183dd25460ab17814637c487534
--- /dev/null
+++ b/140000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5dbb32c69dc2f4ef3dbd674847be461e2229afc321a1c9acfb30e853ead42f64
+size 1445128798
diff --git a/145000/config.json b/145000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/145000/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
+}
diff --git a/145000/diffusion_flax_model.msgpack b/145000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..bbe8518b3da4678e83e5c2d0803d49d64a845268
--- /dev/null
+++ b/145000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21f026a9442594e009e639c5b67bb326ca98a5850e4727b5bfbef7e738fd7913
+size 1445128798
diff --git a/15000/config.json b/15000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/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
+}
diff --git a/15000/diffusion_flax_model.msgpack b/15000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..3babd11d4dc61a9736ba8f01656e6dde17c9d4c9
--- /dev/null
+++ b/15000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb8a18d400f83eed15f8956feaaabbbc0ff9e7a0e091909a7e0837f2bc56a7b5
+size 1445128798
diff --git a/150000/config.json b/150000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/150000/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
+}
diff --git a/150000/diffusion_flax_model.msgpack b/150000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..cf1716bbf0d7f0f0bbcf8383f4e0264330bdff51
--- /dev/null
+++ b/150000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4c29b9e509c0675613cb42da208a063adbfc6761dfbdbb3a6e49d8ee219cb2d
+size 1445128798
diff --git a/155000/config.json b/155000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/155000/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
+}
diff --git a/155000/diffusion_flax_model.msgpack b/155000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d622942b5c3aca853b82d8094497bccc0df1719a
--- /dev/null
+++ b/155000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f17d58ec91e3b852c5488b0bcb163414e965537dc205839484e0e2a19138e835
+size 1445128798
diff --git a/160000/config.json b/160000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/160000/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
+}
diff --git a/160000/diffusion_flax_model.msgpack b/160000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..f8591c41d03818b7379177d4fed5f6ab4d004200
--- /dev/null
+++ b/160000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aea3cfe48d654b6b30aae3de7ab93c2288c1ab0d72892ff4c04c4c524290cdfa
+size 1445128798
diff --git a/165000/config.json b/165000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/165000/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
+}
diff --git a/165000/diffusion_flax_model.msgpack b/165000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..4659efab71ff3cad03404d37cee4794ff186a10d
--- /dev/null
+++ b/165000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cffeb537f79c8f96d203e790b6a1b01b262f4abfdfa94cabea1dc5aa0f587729
+size 1445128798
diff --git a/170000/config.json b/170000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/170000/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
+}
diff --git a/170000/diffusion_flax_model.msgpack b/170000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..8212765cb78109776db35a6aa38a843d31eafcec
--- /dev/null
+++ b/170000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b4d1bcff9c5a56004b0eb0a744e7a682faae3e3e7e73fbfeab816b07027ddab
+size 1445128798
diff --git a/175000/config.json b/175000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/175000/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
+}
diff --git a/175000/diffusion_flax_model.msgpack b/175000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..07ca844674d51080e20c8c0da6714200d2ff0eff
--- /dev/null
+++ b/175000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65dd64a391d2f180aba04542f112f02fdd012c00531e02c2c4fe516d854fdee6
+size 1445128798
diff --git a/180000/config.json b/180000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/180000/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
+}
diff --git a/180000/diffusion_flax_model.msgpack b/180000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..c73d3bf5cf01fd821144900d34ded9a694e95e19
--- /dev/null
+++ b/180000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06c938f4b837d6f1d2eb4e17ed95a8374162bad0424a04097ae329cb7066c027
+size 1445128798
diff --git a/185000/config.json b/185000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/185000/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
+}
diff --git a/185000/diffusion_flax_model.msgpack b/185000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d96aaed940abe8224a39f77781f7df6611d1a1e7
--- /dev/null
+++ b/185000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01a41d5852b160d9ad5304245c08e9570619370c2f6e417466535a51d3847b97
+size 1445128798
diff --git a/190000/config.json b/190000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/190000/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
+}
diff --git a/190000/diffusion_flax_model.msgpack b/190000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..75dba370caba183ac5e58c6495ffb0972f8d2416
--- /dev/null
+++ b/190000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d9b0536ea7c813189aff9256157f68c4586fe281d1211bfd7acd399d1cc6c7b
+size 1445128798
diff --git a/195000/config.json b/195000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/195000/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
+}
diff --git a/195000/diffusion_flax_model.msgpack b/195000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d624f1f23a0cc7f80c2f8e217814347e01f6f75c
--- /dev/null
+++ b/195000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de5c71a3680c6a98594da721af3a6d4f51fb99673f28d2d26c1d9a5cb22e4d9c
+size 1445128798
diff --git a/20000/config.json b/20000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/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
+}
diff --git a/20000/diffusion_flax_model.msgpack b/20000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..10e40783d307541abb65cf2196839646ebd77189
--- /dev/null
+++ b/20000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7397eb4c64460bdc8ef85137c5e6d5e116dc72f7b6ae55faf33bba9364c8a58
+size 1445128798
diff --git a/200000/config.json b/200000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/200000/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
+}
diff --git a/200000/diffusion_flax_model.msgpack b/200000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d4f205868dfd523386426999c38eb3ac0a07c943
--- /dev/null
+++ b/200000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:10a1dfcb11559dfaf202e9a364d987ab39a965062208513bee705b13856d0be3
+size 1445128798
diff --git a/205000/config.json b/205000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/205000/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
+}
diff --git a/205000/diffusion_flax_model.msgpack b/205000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..80abd6118ee2dfcd9abb4f01cc001074a2d875fc
--- /dev/null
+++ b/205000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:602454669d20c28e050670c2b2258d865393f02d6bca8bd202834a3d4d83eecd
+size 1445128798
diff --git a/210000/config.json b/210000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/210000/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
+}
diff --git a/210000/diffusion_flax_model.msgpack b/210000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..297a29c897efa6d0d748164715d39a73d01e1472
--- /dev/null
+++ b/210000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e9ad5eef5bc764f6b585fa2f1a20071dcd4bf9ab5ef2c431a84ee9ebda4da9c
+size 1445128798
diff --git a/215000/config.json b/215000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/215000/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
+}
diff --git a/215000/diffusion_flax_model.msgpack b/215000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..5c1fab379d80a4243c2641df6f29789da8e68eae
--- /dev/null
+++ b/215000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:33c3cf51ad746f237146c99268d701db598d77fb39f922b1c48c0e926c20cada
+size 1445128798
diff --git a/220000/config.json b/220000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/220000/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
+}
diff --git a/220000/diffusion_flax_model.msgpack b/220000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..87860fec88eb2e192a519b25a829ec33e319c1ac
--- /dev/null
+++ b/220000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f5cac57e8f35a78cc427b8ec4bd5534640e33ab26c03564a8dea594ffdc8176
+size 1445128798
diff --git a/225000/config.json b/225000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/225000/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
+}
diff --git a/225000/diffusion_flax_model.msgpack b/225000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..c550408cd2232d41ae0ae46c6f9ac5ec04aabb2d
--- /dev/null
+++ b/225000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98eb062b70d2fe2ee8db7ac6303f77b981bd40a25e00959092d6877d35ae8ba1
+size 1445128798
diff --git a/230000/config.json b/230000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/230000/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
+}
diff --git a/230000/diffusion_flax_model.msgpack b/230000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..3b80c4b14faf65c0a4fa729dd641e35004067c67
--- /dev/null
+++ b/230000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:107c53046c6fbcd8a363e416c6dba4bd332bc5f1891fda39721c275bfd848633
+size 1445128798
diff --git a/235000/config.json b/235000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/235000/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
+}
diff --git a/235000/diffusion_flax_model.msgpack b/235000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..9f940ef4e97ca3263582e90e9685e5592e6060da
--- /dev/null
+++ b/235000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b288e7d765947df7e3e3cb79b701958fc8d82222221978dbfa31aa3ce1bad88
+size 1445128798
diff --git a/240000/config.json b/240000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/240000/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
+}
diff --git a/240000/diffusion_flax_model.msgpack b/240000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..ca7defdc8de812353cd0cb99449c364eb60febe2
--- /dev/null
+++ b/240000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a238c6c29d17a0ba1d416835873473eadbebd977e9c8262b1af7926b336c776
+size 1445128798
diff --git a/25000/config.json b/25000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/25000/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
+}
diff --git a/25000/diffusion_flax_model.msgpack b/25000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d74d85c7ead08de8bbbfd568858c2d1d7b3e9540
--- /dev/null
+++ b/25000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b02f354572a408f9e3d8c55eac4a40ab87ceb29020a2146b6475074f0297269b
+size 1445128798
diff --git a/30000/config.json b/30000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/30000/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
+}
diff --git a/30000/diffusion_flax_model.msgpack b/30000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..4d34789ffa1b2877cda3e7186eba36ca04f64dc5
--- /dev/null
+++ b/30000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e033dda97935873134d0927b0981ae0418afec46956890aafd57b11d67978f5
+size 1445128798
diff --git a/35000/config.json b/35000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/35000/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
+}
diff --git a/35000/diffusion_flax_model.msgpack b/35000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..d6523e088bb42cbb4ad8b5e8fb0f79643da47e74
--- /dev/null
+++ b/35000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:29688bf53cf7b7696bd8e88b1502d449ed8c9dbe8bf1bcf0469c0a3068c0effb
+size 1445128798
diff --git a/40000/config.json b/40000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/40000/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
+}
diff --git a/40000/diffusion_flax_model.msgpack b/40000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..eb66d0c7441ef7021324281160d3e859938b6b6b
--- /dev/null
+++ b/40000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da7913fd23dc47045f881da7b56f3a5c792bdecf416e52cfd788c7a49ff47bd8
+size 1445128798
diff --git a/45000/config.json b/45000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/45000/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
+}
diff --git a/45000/diffusion_flax_model.msgpack b/45000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..614a6518626db0607d622b3e33d68e94f3904fb4
--- /dev/null
+++ b/45000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e283826ec35269dd5f9a027d775fb8c04afb47e525098ccbcf44efc5a8c2d229
+size 1445128798
diff --git a/5000/config.json b/5000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/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
+}
diff --git a/5000/diffusion_flax_model.msgpack b/5000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..bc878d969644c7b2ecef35079e57711f6bc12200
--- /dev/null
+++ b/5000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b4464f1d656d526a65931015bb8be27ea760f26757e6808c4cb9f2520946797
+size 1445128798
diff --git a/50000/config.json b/50000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/50000/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
+}
diff --git a/50000/diffusion_flax_model.msgpack b/50000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..46fc187d6c0d6b8e5a2a34d57d29875da0341c73
--- /dev/null
+++ b/50000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58890b8eb44442c241d7ef637a98d7ff45239eff4383d33e5983089754de8838
+size 1445128798
diff --git a/55000/config.json b/55000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/55000/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
+}
diff --git a/55000/diffusion_flax_model.msgpack b/55000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..5fb043f2fcf8cf8bdbd5838116a8471973b9aaf8
--- /dev/null
+++ b/55000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27203eab971244edaabddc5d48773361c164edeb46e073164df71180c22f1de8
+size 1445128798
diff --git a/60000/config.json b/60000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/60000/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
+}
diff --git a/60000/diffusion_flax_model.msgpack b/60000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..db44e222cefdd0b74c20de8f083ad504510ce99f
--- /dev/null
+++ b/60000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a511cc9f9ad4eae067934515bbb3ec7676b471317e86a4bccdd1fe801130750
+size 1445128798
diff --git a/65000/config.json b/65000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/65000/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
+}
diff --git a/65000/diffusion_flax_model.msgpack b/65000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..5a3331b2363b3088614e0288d63c97e84f81a40c
--- /dev/null
+++ b/65000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:408a000adc355a498082663aae11082c6f23ec2eb6e19a8ebb5d7152b7f7f57b
+size 1445128798
diff --git a/70000/config.json b/70000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/70000/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
+}
diff --git a/70000/diffusion_flax_model.msgpack b/70000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..e2dde30482199a9d2d15ee4d51302a325bc6261a
--- /dev/null
+++ b/70000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e503b9160ea190ed25340bb740c4dc2011589fe06166bb80d945e8ca3d4674
+size 1445128798
diff --git a/75000/config.json b/75000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/75000/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
+}
diff --git a/75000/diffusion_flax_model.msgpack b/75000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..883553eca362aa77dd5eb189766ec98fe334d88c
--- /dev/null
+++ b/75000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cc31b7ce6abf67ab4bbd9854950c9b598005b754aab21599f11a6e168a86376
+size 1445128798
diff --git a/80000/config.json b/80000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/80000/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
+}
diff --git a/80000/diffusion_flax_model.msgpack b/80000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..71fb06c68119269006b4a7124cb99c5dc2553948
--- /dev/null
+++ b/80000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a8c444f4d8caee2c905dae00cdc68d1d5ad1d992d6d789bfcbf504fc7c2b11c
+size 1445128798
diff --git a/85000/config.json b/85000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/85000/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
+}
diff --git a/85000/diffusion_flax_model.msgpack b/85000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..a6101394ac03c94202aff5231c4abe95832ae2c1
--- /dev/null
+++ b/85000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21dc45506a10e9cbfe9511a96536abbb20dcef2017f6263736833882938145f3
+size 1445128798
diff --git a/90000/config.json b/90000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/90000/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
+}
diff --git a/90000/diffusion_flax_model.msgpack b/90000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..af0f04204024b891f6d3ad3b89aa067eb8427d4d
--- /dev/null
+++ b/90000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77dbd5f2b8d7f9261ec002203df59ed0407d6f707e966c96ea95e0f2c314955f
+size 1445128798
diff --git a/95000/config.json b/95000/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549a5ef2eb48dca5102dbaf23eaba82c93a71c41
--- /dev/null
+++ b/95000/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
+}
diff --git a/95000/diffusion_flax_model.msgpack b/95000/diffusion_flax_model.msgpack
new file mode 100644
index 0000000000000000000000000000000000000000..a232b6652db1e18555c0202ec66dc97369df68c7
--- /dev/null
+++ b/95000/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:460ea545de0252b3e5cdfea793c50b090af7961ddd19d74373d207946a1db7e4
+size 1445128798
diff --git a/README.md b/README.md
index e872ddc945cec4e0f365db32936bafa75917c128..33002e845919fd03ca133d804adce3ceeea04e88 100644
--- a/README.md
+++ b/README.md
@@ -1,11 +1,21 @@
+
 ---
-license: apache-2.0
-datasets:
-- keshan/amateur_drawings-controlnet-dataset
-library_name: diffusers
+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
-pipeline_tag: text-to-image
+inference: true
 ---
+    
+# controlnet- keshan/controlnet_amateur_drawings
+
+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: A girl with open arms
+![images_0)](./images_0.png)
 
-Controlnet trained with Facebook's [Amateur Drawings dataset](https://ai.facebook.com/blog/ai-dataset-animation-drawings/). 
\ No newline at end of file
diff --git a/diffusion_flax_model.msgpack b/diffusion_flax_model.msgpack
index c342a45adbb9e0a9de745a89911cd7d6e2ac24bc..64c748acb4a08a1b15d6bd510969f6ab36534bb2 100644
--- a/diffusion_flax_model.msgpack
+++ b/diffusion_flax_model.msgpack
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:96c54af46000d1902c8588c6c7795b10624989dc0cdc0547741c997d29a849f9
+oid sha256:e6672261dd69dc26db2d9e961d8c6beb20573618bd37e8f644e160545a17b650
 size 1445128798
diff --git a/image_control.png b/image_control.png
new file mode 100644
index 0000000000000000000000000000000000000000..f83094edab2eabdb53599126d68dcb012107bb1f
Binary files /dev/null and b/image_control.png differ
diff --git a/images_0.png b/images_0.png
new file mode 100644
index 0000000000000000000000000000000000000000..101bcebb1788145a2b48d73313a416bd7df92557
--- /dev/null
+++ b/images_0.png
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f04fb31cec1ebcea43c2b46112d0c5cc32ce971a754d03076e78ccb8f9c58262
+size 1603873