Update README.md

README.md

@@ -81,9 +81,10 @@ image.save("飞流.png")
 
 ```py
 # !pip install git+https://github.com/huggingface/accelerate
-from diffusers import StableDiffusionPipeline
-
-pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1", torch_dtype=torch.float16, device_map="auto")
+import torch
+torch.backends.cudnn.benchmark = True
+pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1", torch_dtype=torch.float16)
+pipe.to('cuda')
 
 prompt = '飞流直下三千尺，油画'
 image = pipe(prompt, guidance_scale=7.5).images[0]