madebyollin commited on
Commit
476f0d5
1 Parent(s): 6ce2f9a

Upload 6 files

Browse files
README.md CHANGED
@@ -1,3 +1,10 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ # 🍰 Tiny AutoEncoder for Stable Diffusion X4 Upscaler
6
+
7
+ [`taesd-x4-upscaler`](https://github.com/madebyollin/taesd) is very tiny autoencoder which uses the same "latent API" as [`stable-diffusion-x4-upscaler`](https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler)'s VAE.
8
+ `taesd-x4-upscaler` is useful for [real-time previewing](https://twitter.com/madebyollin/status/1679356448655163394) of the upsampling process.
9
+
10
+ This repo contains `.safetensors` versions of the `taesd-x4-upscaler` weights.
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderTiny",
3
+ "_diffusers_version": "0.20.0.dev0",
4
+ "act_fn": "relu",
5
+ "decoder_block_out_channels": [
6
+ 64,
7
+ 64,
8
+ 64
9
+ ],
10
+ "encoder_block_out_channels": [
11
+ 64,
12
+ 64,
13
+ 64
14
+ ],
15
+ "force_upcast": false,
16
+ "in_channels": 3,
17
+ "latent_channels": 4,
18
+ "latent_magnitude": 3,
19
+ "latent_shift": 0.5,
20
+ "num_decoder_blocks": [
21
+ 3,
22
+ 3,
23
+ 1
24
+ ],
25
+ "num_encoder_blocks": [
26
+ 1,
27
+ 3,
28
+ 3
29
+ ],
30
+ "out_channels": 3,
31
+ "scaling_factor": 1.0,
32
+ "upsampling_scaling_factor": 2
33
+ }
diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4997811873dd70e4031c587647b1dea3cccda73253c7cf25947bd542ae5456d2
3
+ size 6863717
diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac9c8b5e53d52aa6ad29d071408285c20191b137ddf5072a929b7ccfb1b5e332
3
+ size 6835820
taesd_decoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a124eb06d813d937e912ac7996382c536e3747ca467144ec6f5f7ca75660103
3
+ size 3417164
taesd_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:422208c11fde651337de0ebf9feafe953f76d551cdda9f4a1c1a1456c7a2d050
3
+ size 3417150