Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
57 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
58 |
"CompVis/stable-diffusion-v1-4",
|
59 |
revision="fp16",
|
60 |
-
torch_dtype=torch.float16 if torch.cuda.is_available() else float
|
61 |
use_auth_token=auth_token,
|
62 |
).to(device)
|
63 |
|
|
|
57 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
58 |
"CompVis/stable-diffusion-v1-4",
|
59 |
revision="fp16",
|
60 |
+
torch_dtype=torch.float16 if torch.cuda.is_available() else float,
|
61 |
use_auth_token=auth_token,
|
62 |
).to(device)
|
63 |
|