Fictiverse akhaliq HF staff commited on
Commit
426f391
1 Parent(s): 9e1f0b9

add how to use diffusers section in model card (#6)

Browse files

- add how to use diffusers section in model card (768d24122bb0a896730edf9944dbd22a1d79a230)


Co-authored-by: Ahsen Khaliq <akhaliq@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +21 -0
README.md CHANGED
@@ -6,6 +6,27 @@ This is the fine-tuned Stable Diffusion model trained on Paper Cut images.
6
 
7
  Use **PaperCut** in your prompts.
8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  ### Sample images:
10
  ![PaperCut.jpg](https://s3.amazonaws.com/moonup/production/uploads/1667910351389-635749860725c2f190a76e88.jpeg)
11
  ![PaperCut.jpg](https://s3.amazonaws.com/moonup/production/uploads/1667912285222-635749860725c2f190a76e88.jpeg)
 
6
 
7
  Use **PaperCut** in your prompts.
8
 
9
+ ### 🧨 Diffusers
10
+
11
+ This model can be used just like any other Stable Diffusion model. For more information,
12
+ please have a look at the [Stable Diffusion](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
13
+
14
+ 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]().
15
+
16
+ ```python
17
+ from diffusers import StableDiffusionPipeline
18
+ import torch
19
+
20
+ model_id = "Fictiverse/Stable_Diffusion_PaperCut_Model"
21
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
22
+ pipe = pipe.to("cuda")
23
+
24
+ prompt = "PaperCut R2-D2"
25
+ image = pipe(prompt).images[0]
26
+
27
+ image.save("./R2-D2.png")
28
+ ```
29
+
30
  ### Sample images:
31
  ![PaperCut.jpg](https://s3.amazonaws.com/moonup/production/uploads/1667910351389-635749860725c2f190a76e88.jpeg)
32
  ![PaperCut.jpg](https://s3.amazonaws.com/moonup/production/uploads/1667912285222-635749860725c2f190a76e88.jpeg)