nightfury commited on
Commit
b82f374
1 Parent(s): cd38ffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,12 +63,12 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
63
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
64
  model_id_or_path,
65
  revision="fp16",
66
- torch_dtype=torch.float16,
67
  use_auth_token=auth_token
68
  ).to(device)
69
  #pipe = pipe.to(device)
70
  #self.register_buffer('n_', ...)
71
- print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
72
 
73
  model = CLIPDensePredT(version='ViT-B/16', reduce_dim=16, complex_trans_conv=True)
74
 
 
63
  pipe = StableDiffusionInpaintingPipeline.from_pretrained(
64
  model_id_or_path,
65
  revision="fp16",
66
+ torch_dtype=torch.float,
67
  use_auth_token=auth_token
68
  ).to(device)
69
  #pipe = pipe.to(device)
70
  #self.register_buffer('n_', ...)
71
+ #print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
72
 
73
  model = CLIPDensePredT(version='ViT-B/16', reduce_dim=16, complex_trans_conv=True)
74