You can deploy this model and start earning money today!

qualcomm/
Stable-Diffusion-v2.1

pytorch

generative_ai

android

unconditional-image-generation

2112.10752

other

Views

No views yet

Stable-Diffusion-v2.1: Optimized for Qualcomm Devices

Generates high resolution images from text prompts using a latent diffusion model. This model uses CLIP ViT-L/14 as text encoder, U-Net based latent denoising, and VAE based decoder to generate the final image.

This is based on the implementation of Stable-Diffusion-v2.1 found here. This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the Qualcomm® AI Hub Models library to export with custom configurations. More details on model performance across various devices, can be found here.

Qualcomm AI Hub Models uses Qualcomm AI Hub Workbench to compile, profile, and evaluate this model. Sign up to run these models on a hosted Qualcomm® device.

Deploy to Snapdragon X Elite NPU

Please follow the Stable Diffusion Windows App tutorial to quantize model with custom weights.

Quantize and Deploy Your Own Fine-Tuned Stable Diffusion

Please follow the Quantize Stable Diffusion tutorial to quantize model with custom weights.

Getting Started

There are two ways to deploy this model on your device:

Option 1: Download Pre-Exported Models

Below are pre-exported model assets ready for deployment.

Runtime	Precision	Chipset	SDK Versions	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X2 Elite	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X Elite	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Gen 3 Mobile	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-8275	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-6690	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-9075	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 7 Gen 4 Mobile	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Mobile	QAIRT 2.45, ONNX Runtime 1.27.1	Download
PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	QAIRT 2.45, ONNX Runtime 1.27.1	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® X2 Elite	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® X Elite	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Gen 3 Mobile	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8775P	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-6690	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-9075	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA7255P	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® 7 Gen 4 Mobile	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Mobile	QAIRT 2.45	Download
QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	QAIRT 2.45	Download

For more device-specific assets and performance metrics, visit Stable-Diffusion-v2.1 on Qualcomm® AI Hub.

Option 2: Export with Custom Configurations

Use the Qualcomm® AI Hub Models Python library to compile and export the model with your own:

Custom weights (e.g., fine-tuned checkpoints)
Custom input shapes
Target device and runtime configurations

This option is ideal if you need to customize the model beyond the default configuration provided here.

See our repository for Stable-Diffusion-v2.1 on GitHub for usage instructions.

Model Details

Model Type: Model_use_case.image_generation

Model Stats:

Input: Text prompt to generate image

Performance Summary

Model	Runtime	Precision	Chipset	Inference Time (ms)	Peak Memory Range (MB)	Primary Compute Unit
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X2 Elite	3.765 ms	1 - 1 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X Elite	7.57 ms	418 - 418 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Gen 3 Mobile	5.812 ms	0 - 13 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-8275	7.843 ms	0 - 4 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	8.088 ms	0 - 450 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-9075	8.093 ms	0 - 3 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-X7181	7.57 ms	418 - 418 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-6690	68.699 ms	0 - 11 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-7790	12.137 ms	0 - 10 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-8750	4.933 ms	0 - 13 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Mobile	4.933 ms	0 - 13 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	4.853 ms	0 - 13 MB	NPU
text_encoder	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 7 Gen 4 Mobile	12.137 ms	0 - 10 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X2 Elite	4.097 ms	0 - 0 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X Elite	7.822 ms	0 - 0 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Gen 3 Mobile	5.847 ms	0 - 8 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	8.153 ms	0 - 3 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	17.384 ms	0 - 8 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	8.125 ms	0 - 2 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8775P	8.303 ms	0 - 10 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8650P	8.303 ms	0 - 10 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8255P	8.303 ms	0 - 10 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-9075	8.106 ms	0 - 2 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-X7181	7.822 ms	0 - 0 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-6690	76.452 ms	0 - 6 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-7790	12.125 ms	0 - 6 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-8750	4.923 ms	0 - 9 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA7255P	17.384 ms	0 - 8 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Mobile	4.923 ms	0 - 9 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	4.827 ms	0 - 9 MB	NPU
text_encoder	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 7 Gen 4 Mobile	12.125 ms	0 - 6 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X2 Elite	67.832 ms	1 - 1 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X Elite	157.431 ms	842 - 842 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Gen 3 Mobile	110.648 ms	0 - 13 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-8275	145.816 ms	0 - 4 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	153.236 ms	0 - 898 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-9075	185.656 ms	0 - 3 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-X7181	157.431 ms	842 - 842 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-6690	2372.602 ms	1 - 11 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-7790	213.662 ms	0 - 11 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-8750	102.239 ms	0 - 12 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Mobile	102.239 ms	0 - 12 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	74.057 ms	0 - 12 MB	NPU
unet	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 7 Gen 4 Mobile	213.662 ms	0 - 11 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X2 Elite	67.792 ms	0 - 0 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X Elite	157.756 ms	1 - 1 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Gen 3 Mobile	109.569 ms	0 - 7 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	145.806 ms	0 - 3 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	346.006 ms	0 - 9 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	152.766 ms	0 - 2 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8775P	147.822 ms	0 - 10 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8650P	147.822 ms	0 - 10 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8255P	147.822 ms	0 - 10 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-9075	146.653 ms	0 - 2 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-X7181	157.756 ms	1 - 1 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-6690	2432.37 ms	0 - 6 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-7790	211.218 ms	0 - 6 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-8750	99.234 ms	1 - 10 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA7255P	346.006 ms	0 - 9 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Mobile	99.234 ms	1 - 10 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	66.857 ms	0 - 9 MB	NPU
unet	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 7 Gen 4 Mobile	211.218 ms	0 - 6 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X2 Elite	128.959 ms	4 - 4 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® X Elite	257.386 ms	65 - 65 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Gen 3 Mobile	196.691 ms	3 - 11 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-8275	262.162 ms	3 - 7 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	263.338 ms	3 - 6 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-9075	241.395 ms	3 - 6 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ IQ-X7181	257.386 ms	65 - 65 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-6690	6246.06 ms	1 - 7 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-7790	503.94 ms	2 - 8 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Qualcomm® Dragonwing™ Q-8750	160.078 ms	4 - 15 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Mobile	160.078 ms	4 - 15 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	121.282 ms	3 - 11 MB	NPU
vae	PRECOMPILED_QNN_ONNX	w8a16	Snapdragon® 7 Gen 4 Mobile	503.94 ms	2 - 8 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X2 Elite	129.013 ms	0 - 0 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® X Elite	257.762 ms	0 - 0 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Gen 3 Mobile	195.493 ms	1 - 8 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	262.61 ms	0 - 4 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-8275	713.361 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ QCS8550 (Proxy)	261.684 ms	0 - 2 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8775P	244.684 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8650P	244.684 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA8255P	244.684 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-9075	262.688 ms	0 - 3 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ IQ-X7181	257.762 ms	0 - 0 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-6690	4473.345 ms	2 - 8 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-7790	502.97 ms	0 - 7 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® Dragonwing™ Q-8750	158.914 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Qualcomm® SA7255P	713.361 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Mobile	158.914 ms	0 - 10 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 8 Elite Gen 5 Mobile	120.715 ms	0 - 9 MB	NPU
vae	QNN_CONTEXT_BINARY	w8a16	Snapdragon® 7 Gen 4 Mobile	502.97 ms	0 - 7 MB	NPU

License

The license for the original implementation of Stable-Diffusion-v2.1 can be found here.

References

Community

Join our AI Hub Slack community to collaborate, post questions and learn more about on-device AI.
For questions or feedback please reach out to us.

Usage and Limitations

This model may not be used for or in connection with any of the following applications:

Accessing essential private and public services and benefits;
Administration of justice and democratic processes;
Assessing or recognizing the emotional state of a person;
Biometric and biometrics-based systems, including categorization of persons based on sensitive characteristics;
Education and vocational training;
Employment and workers management;
Exploitation of the vulnerabilities of persons resulting in harmful behavior;
General purpose social scoring;
Law enforcement;
Management and operation of critical infrastructure;
Migration, asylum and border control management;
Predictive policing;
Real-time remote biometric identification in public spaces;
Recommender systems of social media platforms;
Scraping of facial images (from the internet or otherwise); and/or
Subliminal manipulation

qualcomm/Stable-Diffusion-v2.1