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