more steps
Browse files
app.py
CHANGED
|
@@ -18,6 +18,9 @@ def label_func(fn): return path/"labels"/f"{fn.stem}_P{fn.suffix}"
|
|
| 18 |
|
| 19 |
segmodel = load_learner("camvid-512.pkl")
|
| 20 |
|
|
|
|
|
|
|
|
|
|
| 21 |
inpainting_pipeline = AutoPipelineForInpainting.from_pretrained(
|
| 22 |
"runwayml/stable-diffusion-inpainting",
|
| 23 |
revision="fp16",
|
|
|
|
| 18 |
|
| 19 |
segmodel = load_learner("camvid-512.pkl")
|
| 20 |
|
| 21 |
+
if preferred_device == "cuda":
|
| 22 |
+
segmodel = segmodel.to_fp16()
|
| 23 |
+
|
| 24 |
inpainting_pipeline = AutoPipelineForInpainting.from_pretrained(
|
| 25 |
"runwayml/stable-diffusion-inpainting",
|
| 26 |
revision="fp16",
|