Commit ·
541c94a
1
Parent(s): a2a1398
update readme
Browse files
README.md
CHANGED
|
@@ -49,7 +49,7 @@ tform = transforms.Compose([
|
|
| 49 |
[0.48145466, 0.4578275, 0.40821073],
|
| 50 |
[0.26862954, 0.26130258, 0.27577711]),
|
| 51 |
])
|
| 52 |
-
inp = tform(im).to(device)
|
| 53 |
|
| 54 |
out = sd_pipe(inp, guidance_scale=3)
|
| 55 |
out["images"][0].save("result.jpg")
|
|
|
|
| 49 |
[0.48145466, 0.4578275, 0.40821073],
|
| 50 |
[0.26862954, 0.26130258, 0.27577711]),
|
| 51 |
])
|
| 52 |
+
inp = tform(im).to(device).unsqueeze(0)
|
| 53 |
|
| 54 |
out = sd_pipe(inp, guidance_scale=3)
|
| 55 |
out["images"][0].save("result.jpg")
|