Add variants description to model card (#2)
Browse files
README.md
CHANGED
@@ -16,7 +16,21 @@ The model is trained from scratch 550k steps at resolution `256x256` on a subset
|
|
16 |
|
17 |
![image](https://github.com/Stability-AI/stablediffusion/blob/main/assets/stable-samples/txt2img/merged-0003.png?raw=true)
|
18 |
|
19 |
-
These weights here have been converted to Core ML for use on Apple Silicon hardware.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
- Use it with 𧨠[`diffusers`](https://huggingface.co/stabilityai/stable-diffusion-2-base#examples)
|
22 |
- Use it with the [`stablediffusion`](https://github.com/Stability-AI/stablediffusion) repository: download the `512-base-ema.ckpt` [here](https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt).
|
|
|
16 |
|
17 |
![image](https://github.com/Stability-AI/stablediffusion/blob/main/assets/stable-samples/txt2img/merged-0003.png?raw=true)
|
18 |
|
19 |
+
These weights here have been converted to Core ML for use on Apple Silicon hardware.
|
20 |
+
|
21 |
+
There are 4 variants of the Core ML weights:
|
22 |
+
|
23 |
+
```
|
24 |
+
coreml-stable-diffusion-2-base
|
25 |
+
βββ original
|
26 |
+
β βββ compiled # Swift inference, "original" attention
|
27 |
+
β βββ packages # Python inference, "original" attention
|
28 |
+
βββ split_einsum
|
29 |
+
βββ compiled # Swift inference, "split_einsum" attention
|
30 |
+
βββ packages # Python inference, "split_einsum" attention
|
31 |
+
```
|
32 |
+
|
33 |
+
Please, refer to https://huggingface.co/blog/diffusers-coreml for details.
|
34 |
|
35 |
- Use it with 𧨠[`diffusers`](https://huggingface.co/stabilityai/stable-diffusion-2-base#examples)
|
36 |
- Use it with the [`stablediffusion`](https://github.com/Stability-AI/stablediffusion) repository: download the `512-base-ema.ckpt` [here](https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt).
|