Spaces:

StKirill
/

Detection_and_inpainting

Runtime error

StKirill commited on Dec 23, 2023

Commit

ffc34bd

1 Parent(s): e3b8308

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16",
-    torch_dtype=torch.float16,
 ).to(device)
 def predict(image, prompt):

 pipe = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16",
+    #torch_dtype=torch.float16,
 ).to(device)
 def predict(image, prompt):