init commit of untrained vae

vae/config.json

@@ -1,31 +1,32 @@
 {
-  "_class_name": "VQModel",
+  "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.20.2",
-  "_name_or_path": "CompVis/ldm-celebahq-256",
+  "_name_or_path": "bguisard/stable-diffusion-nano-2-1",
   "act_fn": "silu",
   "block_out_channels": [
     128,
     256,
+    512,
     512
   ],
   "down_block_types": [
+    "DownEncoderBlock2D",
     "DownEncoderBlock2D",
     "DownEncoderBlock2D",
     "DownEncoderBlock2D"
   ],
+  "force_upcast": true,
   "in_channels": 3,
-  "latent_channels": 3,
+  "latent_channels": 4,
   "layers_per_block": 2,
   "norm_num_groups": 32,
-  "norm_type": "group",
-  "num_vq_embeddings": 8192,
   "out_channels": 3,
   "sample_size": 128,
   "scaling_factor": 0.18215,
   "up_block_types": [
+    "UpDecoderBlock2D",
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",
     "UpDecoderBlock2D"
-  ],
-  "vq_embed_dim": null
+  ]
 }

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e95e7e515c8748b15e28335ffbbaec2367efb42c7a7e101b6495f629cc67a4f
-size 110668188
+oid sha256:3e4c08995484ee61270175e9e7a072b66a6e4eeb5f0c266667fe1f45b90daf9a
+size 167335342