initializer?
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from PIL import ImageFilter, ImageEnhance, ImageDraw
|
|
6 |
from diffusers.utils import make_image_grid
|
7 |
from tqdm import tqdm
|
8 |
from diffusers import AutoPipelineForInpainting, LCMScheduler, DDIMScheduler
|
9 |
-
from diffusers import
|
10 |
import numpy as np
|
11 |
from PIL import Image
|
12 |
from datetime import datetime
|
@@ -21,7 +21,7 @@ segmodel = load_learner("camvid-512.pkl")
|
|
21 |
if preferred_device == "cuda":
|
22 |
segmodel = segmodel.to_fp16()
|
23 |
|
24 |
-
inpainting_pipeline =
|
25 |
"runwayml/stable-diffusion-inpainting",
|
26 |
variant="fp16",
|
27 |
torch_dtype=preferred_dtype,
|
|
|
6 |
from diffusers.utils import make_image_grid
|
7 |
from tqdm import tqdm
|
8 |
from diffusers import AutoPipelineForInpainting, LCMScheduler, DDIMScheduler
|
9 |
+
from diffusers import StableDiffusionInpaintPipeline, ControlNetModel
|
10 |
import numpy as np
|
11 |
from PIL import Image
|
12 |
from datetime import datetime
|
|
|
21 |
if preferred_device == "cuda":
|
22 |
segmodel = segmodel.to_fp16()
|
23 |
|
24 |
+
inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
|
25 |
"runwayml/stable-diffusion-inpainting",
|
26 |
variant="fp16",
|
27 |
torch_dtype=preferred_dtype,
|