Spaces:
Running
on
Zero
Running
on
Zero
kayfahaarukku
commited on
Commit
•
914bc5d
1
Parent(s):
9395e21
Bringing back `use_safetensors` and set `add_watermarker` back to `False`.
Browse files
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 |
|