pcuenq HF staff commited on
Commit
1908a35
β€’
1 Parent(s): b2a2beb

Add variants description to model card (#2)

Browse files
Files changed (1) hide show
  1. README.md +15 -1
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. Please, refer to https://huggingface.co/blog/diffusers-coreml for details.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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).