nightfury commited on
Commit
0745c4b
1 Parent(s): bb64902

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
59
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
60
  model_id_or_path,
61
  revision="fp16",
62
- torch_dtype=torch.DoubleTensor, #float16
63
  use_auth_token=auth_token
64
  )
65
  #self.register_buffer('n_', ...)
 
59
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
60
  model_id_or_path,
61
  revision="fp16",
62
+ torch_dtype=torch.double, #float16
63
  use_auth_token=auth_token
64
  )
65
  #self.register_buffer('n_', ...)