Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,9 +31,12 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
|
31 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
32 |
model_id_or_path,
|
33 |
#revision="fp16",
|
34 |
-
torch_dtype=torch.
|
35 |
use_auth_token=auth_token
|
36 |
).to(device)
|
|
|
|
|
|
|
37 |
#pipe = pipe.to(device)
|
38 |
#self.register_buffer('n_', ...)
|
39 |
#print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
|
|
|
31 |
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
32 |
model_id_or_path,
|
33 |
#revision="fp16",
|
34 |
+
torch_dtype=torch.float16,
|
35 |
use_auth_token=auth_token
|
36 |
).to(device)
|
37 |
+
|
38 |
+
pipe = pipe.type(torch.HalfTensor)
|
39 |
+
|
40 |
#pipe = pipe.to(device)
|
41 |
#self.register_buffer('n_', ...)
|
42 |
#print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
|