Hugging Face's logo

qualcomm
/
Stable-Diffusion-v2.1

Unconditional Image Generation
PyTorch
generative_ai
android
Model card Files Community
1
qaihm-bot commited on
Commit
8967fdc
·
verified ·
1 Parent(s): dace9e6

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +67 -56
README.md CHANGED
@@ -1,9 +1,8 @@
1
  ---
2
  library_name: pytorch
3
- license: creativeml-openrail-m
4
  tags:
5
  - generative_ai
6
- - quantized
7
  - android
8
  pipeline_tag: unconditional-image-generation
9
 
@@ -27,7 +26,7 @@ More details on model performance across various devices, can be found
27
 
28
  ### Model Details
29
 
30
- - **Model Type:** Image generation
31
  - **Model Stats:**
32
  - Input: Text prompt to generate image
33
  - Text Encoder Number of parameters: 340M
@@ -35,41 +34,53 @@ More details on model performance across various devices, can be found
35
  - VAE Decoder Number of parameters: 83M
36
  - Model size: 1GB
37
 
38
- | Model | Device | Chipset | Target Runtime | Inference Time (ms) | Peak Memory Range (MB) | Precision | Primary Compute Unit | Target Model
39
  |---|---|---|---|---|---|---|---|---|
40
- | TextEncoderQuantizable | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 6.666 ms | 0 - 3 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/TextEncoderQuantizable.so) |
41
- | TextEncoderQuantizable | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 4.647 ms | 0 - 20 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/TextEncoderQuantizable.so) |
42
- | TextEncoderQuantizable | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 4.2 ms | 0 - 15 MB | W8A16 | NPU | Use Export Script |
43
- | TextEncoderQuantizable | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 6.84 ms | 0 - 0 MB | W8A16 | NPU | Use Export Script |
44
- | TextEncoderQuantizable | SA7255P ADP | SA7255P | QNN | 88.113 ms | 0 - 9 MB | W8A16 | NPU | Use Export Script |
45
- | TextEncoderQuantizable | SA8255 (Proxy) | SA8255P Proxy | QNN | 6.62 ms | 0 - 3 MB | W8A16 | NPU | Use Export Script |
46
- | TextEncoderQuantizable | SA8650 (Proxy) | SA8650P Proxy | QNN | 6.654 ms | 0 - 2 MB | W8A16 | NPU | Use Export Script |
47
- | TextEncoderQuantizable | SA8775P ADP | SA8775P | QNN | 7.869 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
48
- | TextEncoderQuantizable | QCS8275 (Proxy) | QCS8275 Proxy | QNN | 88.113 ms | 0 - 9 MB | W8A16 | NPU | Use Export Script |
49
- | TextEncoderQuantizable | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 6.636 ms | 0 - 3 MB | W8A16 | NPU | Use Export Script |
50
- | TextEncoderQuantizable | QCS9075 (Proxy) | QCS9075 Proxy | QNN | 7.869 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
51
- | UnetQuantizable | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 96.977 ms | 0 - 3 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/UnetQuantizable.so) |
52
- | UnetQuantizable | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 69.178 ms | 0 - 17 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/UnetQuantizable.so) |
53
- | UnetQuantizable | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 61.668 ms | 0 - 14 MB | W8A16 | NPU | Use Export Script |
54
- | UnetQuantizable | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 99.461 ms | 0 - 0 MB | W8A16 | NPU | Use Export Script |
55
- | UnetQuantizable | SA7255P ADP | SA7255P | QNN | 1467.935 ms | 0 - 7 MB | W8A16 | NPU | Use Export Script |
56
- | UnetQuantizable | SA8255 (Proxy) | SA8255P Proxy | QNN | 98.746 ms | 0 - 2 MB | W8A16 | NPU | Use Export Script |
57
- | UnetQuantizable | SA8650 (Proxy) | SA8650P Proxy | QNN | 97.177 ms | 1 - 3 MB | W8A16 | NPU | Use Export Script |
58
- | UnetQuantizable | SA8775P ADP | SA8775P | QNN | 110.665 ms | 0 - 8 MB | W8A16 | NPU | Use Export Script |
59
- | UnetQuantizable | QCS8275 (Proxy) | QCS8275 Proxy | QNN | 1467.935 ms | 0 - 7 MB | W8A16 | NPU | Use Export Script |
60
- | UnetQuantizable | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 97.457 ms | 0 - 3 MB | W8A16 | NPU | Use Export Script |
61
- | UnetQuantizable | QCS9075 (Proxy) | QCS9075 Proxy | QNN | 110.665 ms | 0 - 8 MB | W8A16 | NPU | Use Export Script |
62
- | VaeDecoderQuantizable | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 295.307 ms | 0 - 71 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/VaeDecoderQuantizable.so) |
63
- | VaeDecoderQuantizable | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 223.33 ms | 0 - 312 MB | W8A16 | NPU | [Stable-Diffusion-v2.1.so](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/VaeDecoderQuantizable.so) |
64
- | VaeDecoderQuantizable | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 189.418 ms | 0 - 356 MB | W8A16 | NPU | Use Export Script |
65
- | VaeDecoderQuantizable | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 267.095 ms | 0 - 0 MB | W8A16 | NPU | Use Export Script |
66
- | VaeDecoderQuantizable | SA7255P ADP | SA7255P | QNN | 4460.526 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
67
- | VaeDecoderQuantizable | SA8255 (Proxy) | SA8255P Proxy | QNN | 274.71 ms | 0 - 2 MB | W8A16 | NPU | Use Export Script |
68
- | VaeDecoderQuantizable | SA8650 (Proxy) | SA8650P Proxy | QNN | 269.652 ms | 0 - 2 MB | W8A16 | NPU | Use Export Script |
69
- | VaeDecoderQuantizable | SA8775P ADP | SA8775P | QNN | 301.141 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
70
- | VaeDecoderQuantizable | QCS8275 (Proxy) | QCS8275 Proxy | QNN | 4460.526 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
71
- | VaeDecoderQuantizable | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 271.222 ms | 0 - 3 MB | W8A16 | NPU | Use Export Script |
72
- | VaeDecoderQuantizable | QCS9075 (Proxy) | QCS9075 Proxy | QNN | 301.141 ms | 0 - 10 MB | W8A16 | NPU | Use Export Script |
 
 
 
 
 
 
 
 
 
 
 
 
73
 
74
 
75
 
@@ -79,7 +90,7 @@ More details on model performance across various devices, can be found
79
 
80
  Install the package via pip:
81
  ```bash
82
- pip install "qai-hub-models[stable-diffusion-v2-1-quantized]" -f https://qaihub-public-python-wheels.s3.us-west-2.amazonaws.com/index.html
83
  ```
84
 
85
 
@@ -131,30 +142,30 @@ python -m qai_hub_models.models.stable_diffusion_v2_1_quantized.export
131
  Profiling Results
132
  ------------------------------------------------------------
133
  TextEncoderQuantizable
134
- Device : Samsung Galaxy S23 (13)
135
- Runtime : QNN
136
- Estimated inference time (ms) : 6.7
137
- Estimated peak memory usage (MB): [0, 3]
138
- Total # Ops : 787
139
- Compute Unit(s) : NPU (787 ops)
140
 
141
  ------------------------------------------------------------
142
  UnetQuantizable
143
- Device : Samsung Galaxy S23 (13)
144
- Runtime : QNN
145
- Estimated inference time (ms) : 97.0
146
- Estimated peak memory usage (MB): [0, 3]
147
- Total # Ops : 5891
148
- Compute Unit(s) : NPU (5891 ops)
149
 
150
  ------------------------------------------------------------
151
  VaeDecoderQuantizable
152
- Device : Samsung Galaxy S23 (13)
153
- Runtime : QNN
154
- Estimated inference time (ms) : 295.3
155
- Estimated peak memory usage (MB): [0, 71]
156
- Total # Ops : 189
157
- Compute Unit(s) : NPU (189 ops)
158
  ```
159
 
160
 
 
1
  ---
2
  library_name: pytorch
3
+ license: other
4
  tags:
5
  - generative_ai
 
6
  - android
7
  pipeline_tag: unconditional-image-generation
8
 
 
26
 
27
  ### Model Details
28
 
29
+ - **Model Type:** Model_use_case.image_generation
30
  - **Model Stats:**
31
  - Input: Text prompt to generate image
32
  - Text Encoder Number of parameters: 340M
 
34
  - VAE Decoder Number of parameters: 83M
35
  - Model size: 1GB
36
 
37
+ | Model | Precision | Device | Chipset | Target Runtime | Inference Time (ms) | Peak Memory Range (MB) | Primary Compute Unit | Target Model
38
  |---|---|---|---|---|---|---|---|---|
39
+ | TextEncoderQuantizable | w8a16 | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | QNN | 15.92 ms | 0 - 9 MB | NPU | Use Export Script |
40
+ | TextEncoderQuantizable | w8a16 | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | QNN | 6.594 ms | 0 - 2 MB | NPU | Use Export Script |
41
+ | TextEncoderQuantizable | w8a16 | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | QNN | 6.814 ms | 0 - 9 MB | NPU | Use Export Script |
42
+ | TextEncoderQuantizable | w8a16 | SA7255P ADP | Qualcomm® SA7255P | QNN | 15.92 ms | 0 - 9 MB | NPU | Use Export Script |
43
+ | TextEncoderQuantizable | w8a16 | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | QNN | 6.634 ms | 0 - 3 MB | NPU | Use Export Script |
44
+ | TextEncoderQuantizable | w8a16 | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | QNN | 6.813 ms | 0 - 2 MB | NPU | Use Export Script |
45
+ | TextEncoderQuantizable | w8a16 | SA8775P ADP | Qualcomm® SA8775P | QNN | 6.814 ms | 0 - 9 MB | NPU | Use Export Script |
46
+ | TextEncoderQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | QNN | 6.632 ms | 0 - 3 MB | NPU | Use Export Script |
47
+ | TextEncoderQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | ONNX | 6.859 ms | 0 - 386 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
48
+ | TextEncoderQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | QNN | 4.62 ms | 0 - 19 MB | NPU | Use Export Script |
49
+ | TextEncoderQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | ONNX | 5.075 ms | 0 - 14 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
50
+ | TextEncoderQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | QNN | 4.173 ms | 0 - 14 MB | NPU | Use Export Script |
51
+ | TextEncoderQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | ONNX | 4.499 ms | 0 - 13 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
52
+ | TextEncoderQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 6.834 ms | 0 - 0 MB | NPU | Use Export Script |
53
+ | TextEncoderQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 6.823 ms | 378 - 378 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
54
+ | UnetQuantizable | w8a16 | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | QNN | 241.358 ms | 0 - 8 MB | NPU | Use Export Script |
55
+ | UnetQuantizable | w8a16 | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | QNN | 97.192 ms | 0 - 3 MB | NPU | Use Export Script |
56
+ | UnetQuantizable | w8a16 | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | QNN | 92.111 ms | 0 - 8 MB | NPU | Use Export Script |
57
+ | UnetQuantizable | w8a16 | SA7255P ADP | Qualcomm® SA7255P | QNN | 241.358 ms | 0 - 8 MB | NPU | Use Export Script |
58
+ | UnetQuantizable | w8a16 | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | QNN | 97.903 ms | 0 - 3 MB | NPU | Use Export Script |
59
+ | UnetQuantizable | w8a16 | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | QNN | 96.775 ms | 0 - 2 MB | NPU | Use Export Script |
60
+ | UnetQuantizable | w8a16 | SA8775P ADP | Qualcomm® SA8775P | QNN | 92.111 ms | 0 - 8 MB | NPU | Use Export Script |
61
+ | UnetQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | QNN | 97.437 ms | 0 - 5 MB | NPU | Use Export Script |
62
+ | UnetQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | ONNX | 98.551 ms | 0 - 3 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
63
+ | UnetQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | QNN | 69.198 ms | 0 - 19 MB | NPU | Use Export Script |
64
+ | UnetQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | ONNX | 69.745 ms | 0 - 17 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
65
+ | UnetQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | QNN | 61.688 ms | 0 - 14 MB | NPU | Use Export Script |
66
+ | UnetQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | ONNX | 62.663 ms | 0 - 14 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
67
+ | UnetQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 98.945 ms | 0 - 0 MB | NPU | Use Export Script |
68
+ | UnetQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 99.538 ms | 842 - 842 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
69
+ | VaeDecoderQuantizable | w8a16 | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | QNN | 720.835 ms | 1 - 10 MB | NPU | Use Export Script |
70
+ | VaeDecoderQuantizable | w8a16 | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | QNN | 270.663 ms | 0 - 3 MB | NPU | Use Export Script |
71
+ | VaeDecoderQuantizable | w8a16 | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | QNN | 250.403 ms | 0 - 12 MB | NPU | Use Export Script |
72
+ | VaeDecoderQuantizable | w8a16 | SA7255P ADP | Qualcomm® SA7255P | QNN | 720.835 ms | 1 - 10 MB | NPU | Use Export Script |
73
+ | VaeDecoderQuantizable | w8a16 | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | QNN | 271.924 ms | 0 - 3 MB | NPU | Use Export Script |
74
+ | VaeDecoderQuantizable | w8a16 | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | QNN | 274.917 ms | 0 - 2 MB | NPU | Use Export Script |
75
+ | VaeDecoderQuantizable | w8a16 | SA8775P ADP | Qualcomm® SA8775P | QNN | 250.403 ms | 0 - 12 MB | NPU | Use Export Script |
76
+ | VaeDecoderQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | QNN | 269.888 ms | 0 - 3 MB | NPU | Use Export Script |
77
+ | VaeDecoderQuantizable | w8a16 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | ONNX | 269.758 ms | 0 - 66 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
78
+ | VaeDecoderQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | QNN | 203.298 ms | 0 - 20 MB | NPU | Use Export Script |
79
+ | VaeDecoderQuantizable | w8a16 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | ONNX | 204.412 ms | 3 - 22 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
80
+ | VaeDecoderQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | QNN | 193.019 ms | 0 - 15 MB | NPU | Use Export Script |
81
+ | VaeDecoderQuantizable | w8a16 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | ONNX | 188.783 ms | 3 - 17 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
82
+ | VaeDecoderQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 266.283 ms | 0 - 0 MB | NPU | Use Export Script |
83
+ | VaeDecoderQuantizable | w8a16 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 266.925 ms | 63 - 63 MB | NPU | [Stable-Diffusion-v2.1.onnx](https://huggingface.co/qualcomm/Stable-Diffusion-v2.1/blob/main/Stable-Diffusion-v2.1_w8a16.onnx) |
84
 
85
 
86
 
 
90
 
91
  Install the package via pip:
92
  ```bash
93
+ pip install "qai-hub-models[stable-diffusion-v2-1-quantized]"
94
  ```
95
 
96
 
 
142
  Profiling Results
143
  ------------------------------------------------------------
144
  TextEncoderQuantizable
145
+ Device : cs_8275 (ANDROID 14)
146
+ Runtime : QNN
147
+ Estimated inference time (ms) : 15.9
148
+ Estimated peak memory usage (MB): [0, 9]
149
+ Total # Ops : 971
150
+ Compute Unit(s) : npu (971 ops) gpu (0 ops) cpu (0 ops)
151
 
152
  ------------------------------------------------------------
153
  UnetQuantizable
154
+ Device : cs_8275 (ANDROID 14)
155
+ Runtime : QNN
156
+ Estimated inference time (ms) : 241.4
157
+ Estimated peak memory usage (MB): [0, 8]
158
+ Total # Ops : 5783
159
+ Compute Unit(s) : npu (5783 ops) gpu (0 ops) cpu (0 ops)
160
 
161
  ------------------------------------------------------------
162
  VaeDecoderQuantizable
163
+ Device : cs_8275 (ANDROID 14)
164
+ Runtime : QNN
165
+ Estimated inference time (ms) : 720.8
166
+ Estimated peak memory usage (MB): [1, 10]
167
+ Total # Ops : 189
168
+ Compute Unit(s) : npu (189 ops) gpu (0 ops) cpu (0 ops)
169
  ```
170
 
171