kayfahaarukku commited on
Commit
914bc5d
1 Parent(s): 9395e21

Bringing back `use_safetensors` and set `add_watermarker` back to `False`.

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -55,6 +55,8 @@ def load_pipeline(model_name):
55
  vae=vae,
56
  torch_dtype=torch.float16,
57
  custom_pipeline="lpw_stable_diffusion_xl",
 
 
58
  use_auth_token=HF_TOKEN,
59
  )
60
 
 
55
  vae=vae,
56
  torch_dtype=torch.float16,
57
  custom_pipeline="lpw_stable_diffusion_xl",
58
+ use_safetensors=True,
59
+ add_watermarker=False,
60
  use_auth_token=HF_TOKEN,
61
  )
62