Spaces:

nightfury
/

SD-InPainting

Runtime error

nightfury commited on Oct 12, 2022

Commit

d189791

1 Parent(s): 1ff4e1e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
     model_id_or_path,
     revision="fp16",
-    torch_dtype=torch.double,
     use_auth_token=auth_token
 ).to(device)
 #pipe = pipe.to(device)

 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
     model_id_or_path,
     revision="fp16",
+    torch_dtype=torch.long,
     use_auth_token=auth_token
 ).to(device)
 #pipe = pipe.to(device)