Commit
·
351e642
1
Parent(s):
9a4eb6f
Update README.md
Browse files
README.md
CHANGED
|
@@ -52,7 +52,7 @@ mask_image = download_image(mask_url).resize((512, 512))
|
|
| 52 |
example_image = download_image(example_url).resize((512, 512))
|
| 53 |
|
| 54 |
pipe = DiffusionPipeline.from_pretrained(
|
| 55 |
-
"
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
)
|
| 58 |
pipe = pipe.to("cuda")
|
|
|
|
| 52 |
example_image = download_image(example_url).resize((512, 512))
|
| 53 |
|
| 54 |
pipe = DiffusionPipeline.from_pretrained(
|
| 55 |
+
"Fantasy-Studio/Paint-by-Example",
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
)
|
| 58 |
pipe = pipe.to("cuda")
|