ameerazam08
commited on
Commit
•
ace5058
1
Parent(s):
7e690cc
Update README.md
Browse filesstabilityai/stable-diffusion-xl-base-1-0 to stabilityai/stable-diffusion-xl-base-1.0
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
|