Spaces:

lsb
/

ban-cars

Paused

lsb commited on Mar 9

Commit

3b67ecf

•

1 Parent(s): e674f0f

from_pretrained

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def label_func(fn): return path/"labels"/f"{fn.stem}_P{fn.suffix}"
 segmodel = load_learner("camvid-512.pkl")
-inpainting_pipeline = AutoPipelineForInpainting(
     model="runwayml/stable-diffusion-inpainting",
     revision="fp16",
     torch_dtype=preferred_dtype,

 segmodel = load_learner("camvid-512.pkl")
+inpainting_pipeline = AutoPipelineForInpainting.from_pretrained(
     model="runwayml/stable-diffusion-inpainting",
     revision="fp16",
     torch_dtype=preferred_dtype,