diff --git a/config.json b/config.json
new file mode 100644
index 0000000..0d6aca4
--- /dev/null
+++ b/config.json
@@ -0,0 +1,30 @@
+{
+  "_class_name": "FlaxAutoencoderKL",
+  "_diffusers_version": "0.15.0.dev0",
+  "act_fn": "silu",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "down_block_types": [
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
+  ],
+  "in_channels": 3,
+  "latent_channels": 4,
+  "layers_per_block": 2,
+  "norm_num_groups": 32,
+  "out_channels": 3,
+  "sample_size": 256,
+  "scaling_factor": 0.18215,
+  "up_block_types": [
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D"
+  ]
+}
diff --git a/diffusion_flax_model.msgpack b/diffusion_flax_model.msgpack
new file mode 100644
index 0000000..7661de0
--- /dev/null
+++ b/diffusion_flax_model.msgpack
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f45517affbc93f169d2f5022c6c7d6c7912477f571e8011ddddeca115001562
+size 334623853