File size: 1,047 Bytes
f99d5f3 ca671f8 f99d5f3 ca671f8 f99d5f3 ca671f8 f99d5f3 ca671f8 f99d5f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
license: mit
pipeline_tag: text-to-image
tags:
- onnx
- text-to-image
inference: false
---
## Model Descriptions:
This repo contains ONNX model files for [madebyollin's Tiny AutoEncoder for Stable Diffusion](https://huggingface.co/madebyollin/taesd).
## Using in 🧨 diffusers
To install the requirements for this demo, do pip install optimum["onnxruntime"].
```python
from huggingface_hub import snapshot_download
from diffusers.pipelines import OnnxRuntimeModel
from optimum.onnxruntime import ORTStableDiffusionPipeline
model_id = "CompVis/stable-diffusion-v1-4"
taesd_dir = snapshot_download(repo_id="deinferno/taesd-onnx")
pipeline = ORTStableDiffusionPipeline.from_pretrained(
model_id,
vae_decoder_session = OnnxRuntimeModel.from_pretrained(f"{taesd_dir}/vae_decoder"),
vae_encoder_session = OnnxRuntimeModel.from_pretrained(f"{taesd_dir}/vae_encoder"),
revision="onnx")
prompt = "sailing ship in storm by Leonardo da Vinci"
image = pipeline(prompt).images[0]
image.save("result.png")
```
|