patrickvonplaten commited on
Commit
7044eb4
1 Parent(s): dae081a

Use official weights

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from torchvision import transforms
15
  from diffusers import DiffusionPipeline
16
  from diffusers.utils import torch_device
17
  pipe = DiffusionPipeline.from_pretrained(
18
- "patrickvonplaten/new_inpaint_test",
19
  torch_dtype=torch.float16,
20
  )
21
  pipe = pipe.to("cuda")
 
15
  from diffusers import DiffusionPipeline
16
  from diffusers.utils import torch_device
17
  pipe = DiffusionPipeline.from_pretrained(
18
+ "Fantasy-Studio/Paint-by-Example",
19
  torch_dtype=torch.float16,
20
  )
21
  pipe = pipe.to("cuda")