Hugging Face's logo

Spaces:
ShreeKanade07
/
Sketch_Inpainting
Runtime error

App Files Community
1
ShreeKanade07 commited on
Commit
678836a
•
1 Parent(s): 641ec8a

float16 to float32

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  from diffusers import StableDiffusionInpaintPipeline
18
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
19
- "ShreeKanade07/Real-Image-pipeline", torch_dtype=torch.float16
20
  )
21
  #pipe = pipe.to("cuda")
22
 
 
16
 
17
  from diffusers import StableDiffusionInpaintPipeline
18
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
19
+ "ShreeKanade07/Real-Image-pipeline", torch_dtype=torch.float32
20
  )
21
  #pipe = pipe.to("cuda")
22