KHongJae commited on
Commit
e4b21b6
1 Parent(s): 99eaab5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -32,7 +32,22 @@ DreamBooth for the text encoder was enabled: True.
32
  #### How to use
33
 
34
  ```python
35
- # TODO: add an example code snippet for running this diffusion pipeline
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
36
  ```
37
 
38
  #### Limitations and bias
 
32
  #### How to use
33
 
34
  ```python
35
+ pipeline = DiffusionPipeline.from_pretrained(
36
+ "KHongJae/Chatting_Based_Emoji_Generation_Model", torch_dtype=torch.float16, safety_checker=None
37
+ ).to("cuda")
38
+ pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
39
+ prompt = "Create your own prompt"
40
+ negative_prompt = "Create your own negative prompt"
41
+
42
+ pipeline(
43
+ prompt=prompt,
44
+ negative_prompt=negative_prompt,
45
+ width=200,
46
+ height=200,
47
+ num_inference_steps=50,
48
+ num_images_per_prompt=1,
49
+ generator=torch.manual_seed(123456789),
50
+ ).images[0]
51
  ```
52
 
53
  #### Limitations and bias