fix fp16
Browse files
README.md
CHANGED
|
@@ -91,6 +91,7 @@ image.save("小桥.png")
|
|
| 91 |
|
| 92 |
```py
|
| 93 |
# !pip install git+https://github.com/huggingface/accelerate
|
|
|
|
| 94 |
import torch
|
| 95 |
torch.backends.cudnn.benchmark = True
|
| 96 |
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1", torch_dtype=torch.float16)
|
|
|
|
| 91 |
|
| 92 |
```py
|
| 93 |
# !pip install git+https://github.com/huggingface/accelerate
|
| 94 |
+
from diffusers import StableDiffusionPipeline
|
| 95 |
import torch
|
| 96 |
torch.backends.cudnn.benchmark = True
|
| 97 |
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1", torch_dtype=torch.float16)
|