aykamko commited on
Commit
f5f2a4b
1 Parent(s): 841dec3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -5
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=5.0` is a good default for this scheduler.
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=7).images[0]
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