Spaces:
Runtime error
Runtime error
liuyizhang
commited on
Commit
·
e4af7ee
1
Parent(s):
7273b98
update app.py
Browse files
app.py
CHANGED
|
@@ -250,9 +250,9 @@ logger.info(f"initialize stable-diffusion-inpainting...")
|
|
| 250 |
sd_pipe = None
|
| 251 |
if os.environ.get('IS_MY_DEBUG') is None:
|
| 252 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 253 |
-
|
| 254 |
# revision="fp16"
|
| 255 |
-
"stabilityai/stable-diffusion-2-inpainting",
|
| 256 |
torch_dtype=torch.float16
|
| 257 |
)
|
| 258 |
sd_pipe = sd_pipe.to(device)
|
|
|
|
| 250 |
sd_pipe = None
|
| 251 |
if os.environ.get('IS_MY_DEBUG') is None:
|
| 252 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 253 |
+
"runwayml/stable-diffusion-inpainting",
|
| 254 |
# revision="fp16"
|
| 255 |
+
# "stabilityai/stable-diffusion-2-inpainting",
|
| 256 |
torch_dtype=torch.float16
|
| 257 |
)
|
| 258 |
sd_pipe = sd_pipe.to(device)
|