patrickvonplaten
commited on
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")
|