Upload folder using huggingface_hub

config.json

@@ -1,9 +1,13 @@
 {
-  "image_size": 256,
-  "in_channels": 3,
-  "out_channels": 3,
-  "hidden_channels": 128,
-  "num_res_blocks": 3,
+  "_class_name": "VAEWrapper",
+  "_diffusers_version": "0.33.1",
+  "act_fn": "silu",
+  "attention_resolutions": [
+    32
+  ],
+  "block_out_channels": [
+    64
+  ],
   "channel_multipliers": [
     1,
     2,
@@ -11,19 +15,29 @@
     4,
     4
   ],
-  "attention_resolutions": [
-    32
-  ],
-  "dropout": 0.0,
-  "use_checkpoint": false,
-  "use_fp16": false,
-  "resblock_updown": true,
-  "z_channels": 32,
   "double_z": true,
-  "λ_recon": 1.0,
-  "λ_kl": 1e-6,
-  "λ_lpips": 1.0,
-  "λ_id": 0.1,
-  "λ_gan": 0.5,
-  "gan_start_step": 50000
-}
+  "down_block_types": [
+    "DownEncoderBlock2D"
+  ],
+  "force_upcast": true,
+  "hidden_channels": 128,
+  "image_size": 256,
+  "in_channels": 3,
+  "latent_channels": 4,
+  "latents_mean": null,
+  "latents_std": null,
+  "layers_per_block": 1,
+  "mid_block_add_attention": true,
+  "norm_num_groups": 32,
+  "num_res_blocks": 3,
+  "out_channels": 3,
+  "sample_size": 32,
+  "scaling_factor": 0.18215,
+  "shift_factor": null,
+  "up_block_types": [
+    "UpDecoderBlock2D"
+  ],
+  "use_post_quant_conv": true,
+  "use_quant_conv": true,
+  "z_channels": 32
+}

diffusion_pytorch_model.safetensors

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