Upload 5 files

README.md

@@ -1,3 +1,35 @@
----
-license: apache-2.0
----
+---
+license: mit
+---
+
+# 🍰 Tiny AutoEncoder for FLUX.1
+
+[TAEF1](https://github.com/madebyollin/taesd) is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE.
+FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.
+
+This repo contains `.safetensors` versions of the TAEF1 weights.
+
+## Using in 🧨 diffusers
+
+```python
+import torch
+from diffusers import FluxPipeline, AutoencoderTiny
+
+pipe = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
+)
+pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16)
+pipe.enable_sequential_cpu_offload()
+
+prompt = "slice of delicious New York-style berry cheesecake"
+image = pipe(
+    prompt,
+    guidance_scale=0.0,
+    num_inference_steps=4,
+    max_sequence_length=256,
+).images[0]
+image.save("cheesecake.png")
+```
+
+
+![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/630447d40547362a22a969a2/5FbkYmKc4JsULnskQmzLe.jpeg)

config.json

@@ -0,0 +1,45 @@
+{
+  "_class_name": "AutoencoderTiny",
+  "_diffusers_version": "0.30.0",
+  "act_fn": "relu",
+  "block_out_channels": [
+    64,
+    64,
+    64,
+    64
+  ],
+  "decoder_block_out_channels": [
+    64,
+    64,
+    64,
+    64
+  ],
+  "encoder_block_out_channels": [
+    64,
+    64,
+    64,
+    64
+  ],
+  "force_upcast": false,
+  "in_channels": 3,
+  "latent_channels": 16,
+  "latent_magnitude": 3,
+  "latent_shift": 0.5,
+  "num_decoder_blocks": [
+    3,
+    3,
+    3,
+    1
+  ],
+  "num_encoder_blocks": [
+    1,
+    3,
+    3,
+    3
+  ],
+  "out_channels": 3,
+  "scaling_factor": 1.0,
+  "shift_factor": 0.0,
+  "upsample_fn": "nearest",
+  "upsampling_scaling_factor": 2
+}

diffusion_pytorch_model.bin

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

diffusion_pytorch_model.safetensors

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