johnowhitaker commited on
Commit
124b91d
1 Parent(s): 917d1da

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -0
README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ tags:
4
+ - stable-diffusion
5
+ - stable-diffusion-diffusers
6
+ - text-to-image
7
+ inference: true
8
+ ---
9
+
10
+ To use the mode for inference, just load it like a normal stable diffusion pipeline:
11
+
12
+ ```python
13
+ from diffusers import StableDiffusionPipeline
14
+
15
+ model_path = "johnowhitaker/rainbowdiffusion"
16
+ pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
17
+ pipe.to("cuda")
18
+
19
+ image = pipe(prompt="A cat").images[0]
20
+ image
21
+ ```