update scheduler setting

README.md

@@ -50,13 +50,24 @@ This model can be used for entertainment purposes and as a generative art assist
 ```python
 import torch
 from torch import autocast
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, DDIMScheduler
 
 model_id = "hakurei/waifu-diffusion"
 device = "cuda"
 
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16')
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id,
+    torch_dtype=torch.float16,
+    revision="fp16",
+    scheduler=DDIMScheduler(
+        beta_start=0.00085,
+        beta_end=0.012,
+        beta_schedule="scaled_linear",
+        clip_sample=False,
+        set_alpha_to_one=False,
+    ),
+)
 pipe = pipe.to(device)
 
 prompt = "touhou hakurei_reimu 1girl solo portrait"