KHongJae commited on
Commit
c89565c
1 Parent(s): 25e9472

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -8
README.md CHANGED
@@ -33,19 +33,21 @@ DreamBooth for the text encoder was enabled: True.
33
 
34
  ```python
35
  pipeline = DiffusionPipeline.from_pretrained(
36
- "KHongJae/Chatting_Based_Emoji_Generation_Model", torch_dtype=torch.float16).to("cuda")
 
 
37
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
38
  prompt = "Create your own prompt"
39
  negative_prompt = "Create your own negative prompt"
40
 
41
  pipeline(
42
- prompt=prompt,
43
- negative_prompt=negative_prompt,
44
- width=200,
45
- height=200,
46
- num_inference_steps=50,
47
- num_images_per_prompt=1,
48
- generator=torch.manual_seed(123456789),
49
  ).images[0]
50
  ```
51
 
 
33
 
34
  ```python
35
  pipeline = DiffusionPipeline.from_pretrained(
36
+ "KHongJae/Chatting_Based_Emoji_Generation_Model",
37
+ torch_dtype=torch.float16
38
+ ).to("cuda")
39
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
40
  prompt = "Create your own prompt"
41
  negative_prompt = "Create your own negative prompt"
42
 
43
  pipeline(
44
+ prompt=prompt,
45
+ negative_prompt=negative_prompt,
46
+ width=200,
47
+ height=200,
48
+ num_inference_steps=50,
49
+ num_images_per_prompt=1,
50
+ generator=torch.manual_seed(123456789),
51
  ).images[0]
52
  ```
53