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)
|