tags: | |
- neuron | |
- optimized | |
- aws-neuron | |
- text-to-image | |
base_model: CompVis/stable-diffusion-v1-4 | |
# Neuron-Optimized CompVis/stable-diffusion-v1-4 | |
This repository contains AWS Neuron-optimized files for [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4). | |
## Model Details | |
- **Base Model**: [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4) | |
- **Task**: text-to-image | |
- **Optimization**: AWS Neuron compilation | |
- **Generated by**: [badaoui](https://huggingface.co/badaoui) | |
- **Generated using**: [Optimum Neuron Compiler Space](https://huggingface.co/spaces/optimum/neuron-export) | |
## Usage | |
This model has been optimized for AWS Neuron devices (Inferentia/Trainium). To use it: | |
```python | |
from optimum.neuron import NeuronStableDiffusionPipeline | |
model = NeuronStableDiffusionPipeline.from_pretrained("badaoui/CompVis-stable-diffusion-v1-4-neuron") | |
``` | |
## Performance | |
These files are pre-compiled for AWS Neuron devices and should provide improved inference performance compared to the original model when deployed on Inferentia or Trainium instances. | |
## Original Model | |
For the original model, training details, and more information, please visit: [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4) | |