riccardogiorato commited on
Commit
c53eb8b
1 Parent(s): 065f184

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -27,6 +27,24 @@ By [riccardogiorato](https://twitter.com/riccardogiorato)
27
  >
28
  > **Don't use** the **avatar** keyword, because it's already used by the original model but full of messy data.
29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  # **👇Model👇**
31
 
32
  AI Model Weights available at huggingface: https://huggingface.co/riccardogiorato/avatar-diffusion
 
27
  >
28
  > **Don't use** the **avatar** keyword, because it's already used by the original model but full of messy data.
29
 
30
+ ### 🧨 Diffusers
31
+
32
+ This model can be used just like any other Stable Diffusion model. For more information,
33
+ please have a look at the [Stable Diffusion](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
34
+
35
+ 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]().
36
+
37
+ ```python
38
+ from diffusers import StableDiffusionPipeline
39
+ import torch
40
+ model_id = "riccardogiorato/avatar-diffusion"
41
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
42
+ pipe = pipe.to("cuda")
43
+ prompt = "a magical witch with blue hair with avatartwow style"
44
+ image = pipe(prompt).images[0]
45
+ image.save("./magical_witch.png")
46
+ ```
47
+
48
  # **👇Model👇**
49
 
50
  AI Model Weights available at huggingface: https://huggingface.co/riccardogiorato/avatar-diffusion