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

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:e347df706bdab8d750108e0b5041be1f898684cd273f5e65a4254a789cb73203
+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- Nahrawy/controlnet-vidit
+
+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 building with a stone wall and a cliff, light temperature 4500, light direction NW
+![images_0)](./images_0.png)
+prompt: a wooden table in front of a brick building, light temperature 5500, light direction SE
+![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:387d463b0259825e5c482c1cebb019635986d493572f0f5af7ccdf0867659b65
+size 1445128798