Spaces:
Runtime error
Runtime error
Commit
·
07e4d15
1
Parent(s):
f18d8b0
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,6 @@ def improve_mask(mask):
|
|
| 140 |
vae, unet, tokenizer, text_encoder, scheduler = load_artifacts()
|
| 141 |
pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 142 |
"runwayml/stable-diffusion-inpainting",
|
| 143 |
-
revision="fp32",
|
| 144 |
torch_dtype=torch.float32,
|
| 145 |
use_auth_token=auth_token
|
| 146 |
).to(device)
|
|
|
|
| 140 |
vae, unet, tokenizer, text_encoder, scheduler = load_artifacts()
|
| 141 |
pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 142 |
"runwayml/stable-diffusion-inpainting",
|
|
|
|
| 143 |
torch_dtype=torch.float32,
|
| 144 |
use_auth_token=auth_token
|
| 145 |
).to(device)
|