Spaces:
Running
on
T4
Running
on
T4
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)
|