initial commit

README.md

@@ -1,3 +1,27 @@
 ---
 license: openrail
 ---
+
+VAE for anime style checkpoints
+
+Converted pastel-waifu-diffusion.vae.pt in https://huggingface.co/andite/pastel-mix to diffusers format. 
+
+Example usage
+```
+from diffusers import AutoencoderKL, StableDiffusionPipeline
+import torch
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    pretrained_model_name_or_path='andite/anything-v4.0',
+    vae = AutoencoderKL.from_pretrained('lint/anime_vae')
+)
+
+for component in pipe.components.values():
+    if hasattr(component, 'device'):
+        component.to('cuda', torch.float16)
+
+out = pipe('1girl, blue eyes')
+out.images[0]
+
+```
+

config.json

@@ -0,0 +1,31 @@
+{
+  "_class_name": "AutoencoderKL",
+  "_diffusers_version": "0.14.0.dev0",
+  "_name_or_path": "anime_vae",
+  "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"
+  ]
+}

diffusion_pytorch_model.safetensors

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