FQiao commited on
Commit
d2254e8
1 Parent(s): 82cb92a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -1
README.md CHANGED
@@ -31,7 +31,13 @@ DreamBooth for the text encoder was enabled: False.
31
  #### How to use
32
 
33
  ```python
34
- # TODO: add an example code snippet for running this diffusion pipeline
 
 
 
 
 
 
35
  ```
36
 
37
  #### Limitations and bias
 
31
  #### How to use
32
 
33
  ```python
34
+ from diffusers import AutoPipelineForText2Image
35
+ import torch
36
+
37
+ pipeline = AutoPipelineForText2Image.from_pretrained("FQiao/DreamboothCan", torch_dtype=torch.float16).to("cuda")
38
+ prompt = "a photo of can on the desk"
39
+ image = pipeline(prompt).images[0]
40
+ image
41
  ```
42
 
43
  #### Limitations and bias