Commit
•
45ad647
1
Parent(s):
a44206c
Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,7 @@ import torch
|
|
53 |
|
54 |
# load model and scheduler
|
55 |
model_id = "stabilityai/stable-diffusion-x4-upscaler"
|
56 |
-
pipeline = StableDiffusionUpscalePipeline.from_pretrained(model_id,
|
57 |
pipeline = pipeline.to("cuda")
|
58 |
|
59 |
# let's download an image
|
|
|
53 |
|
54 |
# load model and scheduler
|
55 |
model_id = "stabilityai/stable-diffusion-x4-upscaler"
|
56 |
+
pipeline = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
57 |
pipeline = pipeline.to("cuda")
|
58 |
|
59 |
# let's download an image
|