Update README.md
Browse files
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",
|
|
|
|
|
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 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
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 |
|