Paint-by-Example

Runtime error

patrickvonplaten commited on Dec 7, 2022

Commit

806bd4b

•

1 Parent(s): dae081a

Use official weights (#3)

- Use official weights (7044eb4760b766cba4b4105aa6ee504ad3d10c7c)

Co-authored-by: Patrick von Platen <[email protected]>

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from torchvision import transforms
 from diffusers import DiffusionPipeline
 from diffusers.utils import torch_device
 pipe = DiffusionPipeline.from_pretrained(
-    "patrickvonplaten/new_inpaint_test",
     torch_dtype=torch.float16,
 )
 pipe = pipe.to("cuda")

 from diffusers import DiffusionPipeline
 from diffusers.utils import torch_device
 pipe = DiffusionPipeline.from_pretrained(
+    "Fantasy-Studio/Paint-by-Example",
     torch_dtype=torch.float16,
 )
 pipe = pipe.to("cuda")