Update README.md
Browse files
README.md
CHANGED
@@ -37,10 +37,8 @@ pip install transformers accelerate safetensors
|
|
37 |
```
|
38 |
|
39 |
**Notes:**
|
40 |
-
- The pipeline uses the `EDMEulerScheduler` scheduler. It's an [EDM formulation](https://arxiv.org/abs/2206.00364) of the Euler scheduler.
|
41 |
-
- `guidance_scale=
|
42 |
-
- The pipeline also supports the `EDMDPMSolverMultistepScheduler` scheduler. It's an [EDM formulation](https://arxiv.org/abs/2206.00364) of the DPM scheduler.
|
43 |
-
- `guidance_scale=3.0` is a good default for this scheduler.
|
44 |
|
45 |
Then, run the following snippet:
|
46 |
|
@@ -54,8 +52,12 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
54 |
variant="fp16",
|
55 |
).to("cuda")
|
56 |
|
|
|
|
|
|
|
|
|
57 |
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
|
58 |
-
image = pipe(prompt=prompt, num_inference_steps=50, guidance_scale=
|
59 |
```
|
60 |
|
61 |
### Using the model with Automatic1111/ComfyUI
|
|
|
37 |
```
|
38 |
|
39 |
**Notes:**
|
40 |
+
- The pipeline uses the `EDMEulerScheduler` scheduler by default. It's an [EDM formulation](https://arxiv.org/abs/2206.00364) of the Euler scheduler. `guidance_scale=5.0` is a good default for this scheduler.
|
41 |
+
- The pipeline also supports the `EDMDPMSolverMultistepScheduler` scheduler, for crisper fine details. It's an [EDM formulation](https://arxiv.org/abs/2206.00364) of the DPM++ 2M Karras scheduler. `guidance_scale=3.0` is a good default for this scheduler.
|
|
|
|
|
42 |
|
43 |
Then, run the following snippet:
|
44 |
|
|
|
52 |
variant="fp16",
|
53 |
).to("cuda")
|
54 |
|
55 |
+
# # Optional: Use DPM++ 2M Karras scheduler for crisper fine details
|
56 |
+
# from diffusers import EDMDPMSolverMultistepScheduler
|
57 |
+
# pipe.scheduler = EDMDPMSolverMultistepScheduler()
|
58 |
+
|
59 |
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
|
60 |
+
image = pipe(prompt=prompt, num_inference_steps=50, guidance_scale=3).images[0]
|
61 |
```
|
62 |
|
63 |
### Using the model with Automatic1111/ComfyUI
|