yizhangliu
commited on
Commit
•
4a2684c
1
Parent(s):
b4fd607
Update app.py
Browse files
app.py
CHANGED
@@ -284,9 +284,9 @@ def load_sd_model():
|
|
284 |
if os.environ.get('IS_MY_DEBUG') is None:
|
285 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
286 |
"runwayml/stable-diffusion-inpainting",
|
287 |
-
|
288 |
# "stabilityai/stable-diffusion-2-inpainting",
|
289 |
-
|
290 |
)
|
291 |
sd_pipe = sd_pipe.to(device)
|
292 |
|
|
|
284 |
if os.environ.get('IS_MY_DEBUG') is None:
|
285 |
sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
286 |
"runwayml/stable-diffusion-inpainting",
|
287 |
+
revision="fp16",
|
288 |
# "stabilityai/stable-diffusion-2-inpainting",
|
289 |
+
torch_dtype=torch.float16,
|
290 |
)
|
291 |
sd_pipe = sd_pipe.to(device)
|
292 |
|