Spaces:
Runtime error
Runtime error
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)
|