File size: 1,181 Bytes
282a727
6db72ca
282a727
6db72ca
 
 
 
 
 
 
 
282a727
6db72ca
f22c0e9
6db72ca
 
 
f22c0e9
 
 
6db72ca
f22c0e9
6db72ca
 
 
 
 
 
 
 
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
---
pipeline_tag: text-to-image
license: creativeml-openrail-m
base_model: runwayml/stable-diffusion-v1-5
language:
  - en
tags:
  - stable-diffusion
  - onnxruntime
  - onnx
  - text-to-image
---

# 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](https://github.com/microsoft/Olive/tree/main/examples/stable_diffusion) 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](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
- **Model Description:** This is a conversion of the [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) model for [ONNX Runtime](https://github.com/microsoft/onnxruntime) inference with CUDA execution provider.