Update README.md
#5
by ameerazam08 - opened
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
|