nightfury commited on
Commit
60e887d
1 Parent(s): e9d5a30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
57
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
58
  "CompVis/stable-diffusion-v1-4",
59
  revision="fp16",
60
- torch_dtype=torch.float16 if torch.cuda.is_available() else float
61
  use_auth_token=auth_token,
62
  ).to(device)
63
 
 
57
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
58
  "CompVis/stable-diffusion-v1-4",
59
  revision="fp16",
60
+ torch_dtype=torch.float16 if torch.cuda.is_available() else float,
61
  use_auth_token=auth_token,
62
  ).to(device)
63