nightfury commited on
Commit
3a4042f
1 Parent(s): 002e501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -31,9 +31,12 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
31
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
32
  model_id_or_path,
33
  #revision="fp16",
34
- torch_dtype=torch.float,
35
  use_auth_token=auth_token
36
  ).to(device)
 
 
 
37
  #pipe = pipe.to(device)
38
  #self.register_buffer('n_', ...)
39
  #print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
31
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
32
  model_id_or_path,
33
  #revision="fp16",
34
+ torch_dtype=torch.float16,
35
  use_auth_token=auth_token
36
  ).to(device)
37
+
38
+ pipe = pipe.type(torch.HalfTensor)
39
+
40
  #pipe = pipe.to(device)
41
  #self.register_buffer('n_', ...)
42
  #print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())