Commit
•
c9822f0
1
Parent(s):
c35c0a0
Update README.md (#5)
Browse files- Update README.md (ace5058aafb5f81d4d9f60b8db0d449c626d7aea)
Co-authored-by: Ameer Azam <[email protected]>
README.md
CHANGED
|
@@ -63,7 +63,7 @@ You can use the model then as follows
|
|
| 63 |
from diffusers import DiffusionPipeline
|
| 64 |
import torch
|
| 65 |
|
| 66 |
-
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1
|
| 67 |
pipe.to("cuda")
|
| 68 |
|
| 69 |
# if using torch < 2.0
|
|
|
|
| 63 |
from diffusers import DiffusionPipeline
|
| 64 |
import torch
|
| 65 |
|
| 66 |
+
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
|
| 67 |
pipe.to("cuda")
|
| 68 |
|
| 69 |
# if using torch < 2.0
|