carsonkatri commited on
Commit
33780c8
1 Parent(s): adc2b43

Add example code

Browse files
Files changed (1) hide show
  1. README.md +26 -0
README.md CHANGED
@@ -1,3 +1,29 @@
1
  ---
2
  license: openrail
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: openrail
3
  ---
4
+
5
+ This model is converted from [stable-diffusion-2-depth](https://huggingface.co/stabilityai/stable-diffusion-2-depth) using the [conversion script](https://github.com/huggingface/diffusers/blob/main/scripts/convert_original_stable_diffusion_to_diffusers.py) from [🤗 Diffusers](https://github.com/huggingface/diffusers).
6
+
7
+ You can use it with the pipeline here: https://gist.github.com/carson-katri/f51532b9d5162928d5cacbaee081a799
8
+
9
+ ```python
10
+ # class StableDiffusionDepthPipeline: ...
11
+
12
+ from PIL import Image
13
+
14
+ model_id = "carsonkatri/stable-diffusion-2-depth-diffusers"
15
+
16
+ # Use the pipeline from this GH Gist: https://gist.github.com/carson-katri/f51532b9d5162928d5cacbaee081a799
17
+ pipe = StableDiffusionDepthPipeline.from_pretrained(model_id)
18
+ pipe = pipe.to("cuda")
19
+
20
+ image = pipe(
21
+ prompt="a photo of a stormtrooper from star wars",
22
+ depth_image=Image.open('depth.png'), # Black and white depth map
23
+ image=Image.open("emad.png"), # Optional init image and strength.
24
+ width=768,
25
+ height=512
26
+ ).images[0]
27
+
28
+ image.save('stormtrooper.png')
29
+ ```