End of training

.gitattributes

@@ -33,3 +33,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

README.md

@@ -0,0 +1,45 @@
+---
+base_model: black-forest-labs/FLUX.1-dev
+library_name: diffusers
+license: openrail++
+tags:
+- flux
+- flux-diffusers
+- text-to-image
+- diffusers
+- controlnet
+- diffusers-training
+inference: true
+---
+
+<!-- This model card has been generated automatically according to the information the training script had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+
+# controlnet-laidawang/test_flux_controlnet
+
+These are controlnet weights trained on black-forest-labs/FLUX.1-dev with new type of conditioning.
+You can find some example images below.
+
+prompt: red circle with blue background
+![images_0)](./images_0.png)
+prompt: cyan circle with brown floral background
+![images_1)](./images_1.png)
+
+
+
+## Intended uses & limitations
+
+#### How to use
+
+```python
+# TODO: add an example code snippet for running this diffusion pipeline
+```
+
+#### Limitations and bias
+
+[TODO: provide examples of latent issues and potential remediations]
+
+## Training details
+
+[TODO: describe the data used to train the model]

checkpoint-2/flux_controlnet/config.json

@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxControlNetModel",
+  "_diffusers_version": "0.31.0.dev0",
+  "_name_or_path": "/data3/home/srchen/PROJECT/flux_controlnet_annotator/models/dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 64,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 4,
+  "num_mode": null,
+  "num_single_layers": 0,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}

checkpoint-2/flux_controlnet/diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:08c305bcdae8238741e080ce5e82e394d956744e8266a977097a261e74f3988c
+size 5769352416

checkpoint-2/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37e3106e70a49ce3c1fef3788684b602a140d412e21c4cbfe766c5ec9e1b5963
+size 10859172202

checkpoint-2/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7da54bcc82a38146adaa8d7e49e0faa882f11acce66cec10e925d28e48b49bc9
+size 14408

checkpoint-2/scheduler.bin

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

config.json

@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxControlNetModel",
+  "_diffusers_version": "0.31.0.dev0",
+  "_name_or_path": "/data3/home/srchen/PROJECT/flux_controlnet_annotator/models/dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 64,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 4,
+  "num_mode": null,
+  "num_single_layers": 0,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}

diffusion_pytorch_model.bf16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d4c54b1d914804e94ed7f075d6593fea97e4571bf8e854756a164df73a94b22
+size 2884685112