Richard Neuschulz
commited on
Commit
·
b0583c8
1
Parent(s):
a60f6b6
updated app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,9 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
| 28 |
base_model_path,
|
| 29 |
torch_dtype=torch.float16,
|
| 30 |
scheduler=noise_scheduler,
|
| 31 |
-
add_watermarker=False
|
|
|
|
|
|
|
| 32 |
# vae=vae,
|
| 33 |
#feature_extractor=safety_feature_extractor,
|
| 34 |
#safety_checker=safety_checker
|
|
|
|
| 28 |
base_model_path,
|
| 29 |
torch_dtype=torch.float16,
|
| 30 |
scheduler=noise_scheduler,
|
| 31 |
+
add_watermarker=False,
|
| 32 |
+
use_safetensors=True,
|
| 33 |
+
variant="fp16"
|
| 34 |
# vae=vae,
|
| 35 |
#feature_extractor=safety_feature_extractor,
|
| 36 |
#safety_checker=safety_checker
|