Update README.md
Browse files
README.md
CHANGED
@@ -32,7 +32,22 @@ DreamBooth for the text encoder was enabled: True.
|
|
32 |
#### How to use
|
33 |
|
34 |
```python
|
35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|