yahoo-inc
/

photo-background-generation

StableDiffusionControlNetInpaintPipeline

yahoo-open-source-software-incubator

Model card Files Files and versions Community

erfan-yahoo commited on May 8

Commit

719f1ac

•

1 Parent(s): 16f6e14

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -19,8 +19,8 @@ The paper addresses an issue we call "object expansion" when generating backgrou
 ### Load pipeline
 ```py
 from diffusers import DiffusionPipeline
-pipeline = DiffusionPipeline.from_pretrained("yahoo-inc/photo-background-generation")
 pipeline = pipeline.to('cuda')
 ```

 ### Load pipeline
 ```py
 from diffusers import DiffusionPipeline
+model_id = "yahoo-inc/photo-background-generation"
+pipeline = DiffusionPipeline.from_pretrained(model_id, custom_pipeline=model_id)
 pipeline = pipeline.to('cuda')
 ```