tlwu/stable-diffusion-v1-5-onnxruntime
By tlwu · ↓ 0 · ♥ 4 · Updated 2023-12-31
What's in this repo
This page lists the files in tlwu/stable-diffusion-v1-5-onnxruntime, including model.onnx and more.
- 5 ONNX models (.onnx files)
- 14 files, including 6 .json, 5 .onnx, 1 .md, 1 .txt
- Last updated 2023-12-31
ONNX files
| File | Size | Download |
|---|---|---|
| safety_checker/model.onnx | — | Download |
| text_encoder/model.onnx | — | Download |
| unet/model.onnx | — | Download |
| vae_decoder/model.onnx | — | Download |
| vae_encoder/model.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
Stable Diffusion 1.5 for ONNX Runtime CUDA provider
Introduction
This repository hosts the optimized onnx models of Stable Diffusion 1.5 to accelerate inference with ONNX Runtime CUDA execution provider on Nvidia GPUs. It cannot run in other execution providers like CPU or DirectML.
The models are generated by Olive with command like the following:
python stable_diffusion.py --provider cuda --optimize
Model Description
- Developed by: Stability AI
- Model type: Diffusion-based text-to-image generative model
- License: The CreativeML OpenRAIL M license
- Model Description: This is a conversion of the runwayml/stable-diffusion-v1-5 model for ONNX Runtime inference with CUDA execution provider.
Model info
Created: 2023-12-30
Updated: 2023-12-31
View on Hugging Face