Spaces:

lsb
/

ban-cars

Paused

lsb commited on Mar 9

Commit

7c3d7a0

•

1 Parent(s): 4cc3150

initializer?

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import ImageFilter, ImageEnhance, ImageDraw
 from diffusers.utils import make_image_grid
 from tqdm import tqdm
 from diffusers import AutoPipelineForInpainting, LCMScheduler, DDIMScheduler
-from diffusers import StableDiffusionControlNetInpaintPipeline, ControlNetModel
 import numpy as np
 from PIL import Image
 from datetime import datetime
@@ -21,7 +21,7 @@ segmodel = load_learner("camvid-512.pkl")
 if preferred_device == "cuda":
     segmodel = segmodel.to_fp16()
-inpainting_pipeline = AutoPipelineForInpainting.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     variant="fp16",
     torch_dtype=preferred_dtype,

 from diffusers.utils import make_image_grid
 from tqdm import tqdm
 from diffusers import AutoPipelineForInpainting, LCMScheduler, DDIMScheduler
+from diffusers import StableDiffusionInpaintPipeline, ControlNetModel
 import numpy as np
 from PIL import Image
 from datetime import datetime
 if preferred_device == "cuda":
     segmodel = segmodel.to_fp16()
+inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     variant="fp16",
     torch_dtype=preferred_dtype,