Spaces:

lsb
/

ban-cars

Paused

lsb commited on Mar 9

Commit

b80c840

•

1 Parent(s): 3b67ecf

kwargslol

Files changed (1) hide show

app.py CHANGED Viewed

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

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