WiNE-iNEFF commited on
Commit
9a76916
·
1 Parent(s): 6a3d4af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -2
README.md CHANGED
@@ -5,14 +5,27 @@ tags:
5
  - diffusers
6
  - unconditional-image-generation
7
  - diffusion-model
 
8
  ---
9
 
10
  ## Usage
11
 
12
  ```python
13
  from diffusers import DDPMPipeline
 
 
 
 
 
 
 
 
 
14
 
15
  pipeline = DDPMPipeline.from_pretrained('WiNE-iNEFF/Minecraft-Skin-Diffusion')
16
  image = pipeline().images[0].convert('RGBA')
17
- image
18
- ```
 
 
 
 
5
  - diffusers
6
  - unconditional-image-generation
7
  - diffusion-model
8
+ pipeline_tag: unconditional-image-generation
9
  ---
10
 
11
  ## Usage
12
 
13
  ```python
14
  from diffusers import DDPMPipeline
15
+ import torch
16
+
17
+ device = (
18
+ "mps"
19
+ if torch.backends.mps.is_available()
20
+ else "cuda"
21
+ if torch.cuda.is_available()
22
+ else "cpu"
23
+ )
24
 
25
  pipeline = DDPMPipeline.from_pretrained('WiNE-iNEFF/Minecraft-Skin-Diffusion')
26
  image = pipeline().images[0].convert('RGBA')
27
+
28
+ pipeline_name = "WiNE-iNEFF/Minecraft-Skin-Diffusion"
29
+ image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
30
+ image_pipe(num_inference_steps=40).images[0].save("skin.png")
31
+ ```