nitrosocke patrickvonplaten commited on
Commit
f9ecab4
1 Parent(s): e2c6509

Add diffusers example (#3)

Browse files

- Add diffusers example (6e4dac8a9648a3f9be2643f07c664e2e9620ddc9)


Co-authored-by: Patrick von Platen <[email protected]>

Files changed (1) hide show
  1. README.md +22 -0
README.md CHANGED
@@ -8,6 +8,28 @@ Use the tokens **_archer style_** in your prompts for the effect.
8
 
9
  If you enjoy this model, please check out my other models on [Huggingface](https://huggingface.co/nitrosocke)
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  **Portraits rendered with the model:**
12
  ![Portrait Samples](https://huggingface.co/nitrosocke/archer-diffusion/resolve/main/archer-diffusion-samples.png)
13
  **Celebrities rendered with the model:**
 
8
 
9
  If you enjoy this model, please check out my other models on [Huggingface](https://huggingface.co/nitrosocke)
10
 
11
+ ### 🧨 Diffusers
12
+
13
+ This model can be used just like any other Stable Diffusion model. For more information,
14
+ please have a look at the [Stable Diffusion](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
15
+
16
+ You can also export the model to [ONNX](https://huggingface.co/docs/diffusers/optimization/onnx), [MPS](https://huggingface.co/docs/diffusers/optimization/mps) and/or [FLAX/JAX]().
17
+
18
+ ```python
19
+ #!pip install diffusers transformers scipy torch
20
+ from diffusers import StableDiffusionPipeline
21
+ import torch
22
+
23
+ model_id = "nitrosocke/archer-diffusion"
24
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
25
+ pipe = pipe.to("cuda")
26
+
27
+ prompt = "a magical princess with golden hair, archer style"
28
+ image = pipe(prompt).images[0]
29
+
30
+ image.save("./magical_princess.png")
31
+ ```
32
+
33
  **Portraits rendered with the model:**
34
  ![Portrait Samples](https://huggingface.co/nitrosocke/archer-diffusion/resolve/main/archer-diffusion-samples.png)
35
  **Celebrities rendered with the model:**