erfan-yahoo commited on
Commit
719f1ac
1 Parent(s): 16f6e14

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,8 +19,8 @@ The paper addresses an issue we call "object expansion" when generating backgrou
19
  ### Load pipeline
20
  ```py
21
  from diffusers import DiffusionPipeline
22
-
23
- pipeline = DiffusionPipeline.from_pretrained("yahoo-inc/photo-background-generation")
24
  pipeline = pipeline.to('cuda')
25
  ```
26
 
 
19
  ### Load pipeline
20
  ```py
21
  from diffusers import DiffusionPipeline
22
+ model_id = "yahoo-inc/photo-background-generation"
23
+ pipeline = DiffusionPipeline.from_pretrained(model_id, custom_pipeline=model_id)
24
  pipeline = pipeline.to('cuda')
25
  ```
26