fix fp16

README.md

@@ -82,6 +82,7 @@ image.save("飞流.png")
 ```py
 # !pip install git+https://github.com/huggingface/accelerate
 import torch
+from diffusers import StableDiffusionPipeline
 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')