Spaces:
Running
on
T4
Running
on
T4
liuyizhang
commited on
Commit
•
70c927a
1
Parent(s):
c7d2050
update app.py
Browse files
app.py
CHANGED
@@ -246,7 +246,8 @@ logger.info(f"initialize stable-diffusion-inpainting...")
|
|
246 |
sd_pipe = None
|
247 |
if os.environ.get('IS_MY_DEBUG') is None:
|
248 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
249 |
-
"runwayml/stable-diffusion-inpainting",
|
|
|
250 |
torch_dtype=torch.float16
|
251 |
)
|
252 |
sd_pipe = sd_pipe.to(device)
|
|
|
246 |
sd_pipe = None
|
247 |
if os.environ.get('IS_MY_DEBUG') is None:
|
248 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
249 |
+
# "runwayml/stable-diffusion-inpainting",
|
250 |
+
"stabilityai/stable-diffusion-2-inpainting",
|
251 |
torch_dtype=torch.float16
|
252 |
)
|
253 |
sd_pipe = sd_pipe.to(device)
|