Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
|
63 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
64 |
model_id_or_path,
|
65 |
revision="fp16",
|
66 |
-
torch_dtype=torch.float16
|
67 |
use_auth_token=auth_token
|
68 |
).to(device)
|
69 |
#pipe = pipe.to(device)
|
|
|
63 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
64 |
model_id_or_path,
|
65 |
revision="fp16",
|
66 |
+
torch_dtype=torch.float16,
|
67 |
use_auth_token=auth_token
|
68 |
).to(device)
|
69 |
#pipe = pipe.to(device)
|