madebyollin
commited on
Commit
•
614f768
1
Parent(s):
44c14a1
Update README.md
Browse files
README.md
CHANGED
@@ -7,10 +7,6 @@ license: mit
|
|
7 |
[TAESD](https://github.com/madebyollin/taesd) is very tiny autoencoder which uses the same "latent API" as Stable Diffusion's VAE.
|
8 |
TAESD is useful for [real-time previewing](https://twitter.com/madebyollin/status/1679356448655163394) of the SD generation process.
|
9 |
|
10 |
-
Comparison on my laptop:
|
11 |
-
|
12 |
-
![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/630447d40547362a22a969a2/Xj0l9V2wpG7YWTzrGxpWI.jpeg)
|
13 |
-
|
14 |
This repo contains `.safetensors` versions of the TAESD weights.
|
15 |
|
16 |
For SDXL, use [TAESDXL](https://huggingface.co/madebyollin/taesdxl/) instead (the SD and SDXL VAEs are [incompatible](https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/discussions/6#64b8a9c13707b7d603c6ac16)).
|
@@ -27,7 +23,9 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
27 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd", torch_dtype=torch.float16)
|
28 |
pipe = pipe.to("cuda")
|
29 |
|
30 |
-
prompt = "slice of delicious New York-style
|
31 |
-
image = pipe(prompt, num_inference_steps=
|
32 |
image.save("cheesecake.png")
|
33 |
-
```
|
|
|
|
|
|
7 |
[TAESD](https://github.com/madebyollin/taesd) is very tiny autoencoder which uses the same "latent API" as Stable Diffusion's VAE.
|
8 |
TAESD is useful for [real-time previewing](https://twitter.com/madebyollin/status/1679356448655163394) of the SD generation process.
|
9 |
|
|
|
|
|
|
|
|
|
10 |
This repo contains `.safetensors` versions of the TAESD weights.
|
11 |
|
12 |
For SDXL, use [TAESDXL](https://huggingface.co/madebyollin/taesdxl/) instead (the SD and SDXL VAEs are [incompatible](https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/discussions/6#64b8a9c13707b7d603c6ac16)).
|
|
|
23 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd", torch_dtype=torch.float16)
|
24 |
pipe = pipe.to("cuda")
|
25 |
|
26 |
+
prompt = "slice of delicious New York-style cheesecake topped with berries, mint, chocolate crumble"
|
27 |
+
image = pipe(prompt, num_inference_steps=50, generator=torch.Generator("cpu").manual_seed(0x7A35D)).images[0]
|
28 |
image.save("cheesecake.png")
|
29 |
+
```
|
30 |
+
|
31 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/630447d40547362a22a969a2/m4pVdlJ25U774v04Tsgzu.png)
|