Spaces:

AsmaAILab
/

empty-room-generation

Sleeping

AsmaAILab commited on Jan 23

Commit

d4a6270

verified ·

1 Parent(s): 40a6dc7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ controlnet_depth = ControlNetModel.from_pretrained(
 # Load Realistic Vision model for final realism boost
 realistic_vision = StableDiffusionInpaintPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-2-1", #"sd2-community/stable-diffusion-2-1", #"runwayml/stable-diffusion-inpainting",
     torch_dtype=DTYPE,
 ).to(DEVICE)

 # Load Realistic Vision model for final realism boost
 realistic_vision = StableDiffusionInpaintPipeline.from_pretrained(
+    "Manojb/stable-diffusion-2-1-base",  # "stabilityai/stable-diffusion-2-1", "sd2-community/stable-diffusion-2-1", #"runwayml/stable-diffusion-inpainting",
     torch_dtype=DTYPE,
 ).to(DEVICE)