模型说明：

本仓库包含 madebyollin 的 Stable Diffusion Tiny AutoEncoder（小型自编码器） 的 ONNX 模型文件。

在 🧨 diffusers 中使用

要安装本演示所需依赖，请运行 pip install optimum["onnxruntime"]。

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")