Update app.py
Browse files
app.py
CHANGED
@@ -20,8 +20,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
20 |
torch_dtype=torch.float16,
|
21 |
use_safetensors=True,
|
22 |
custom_pipeline="lpw_stable_diffusion_xl",
|
23 |
-
add_watermarker=False
|
24 |
-
variant="fp16"
|
25 |
)
|
26 |
pipe.to('cuda')
|
27 |
|
|
|
20 |
torch_dtype=torch.float16,
|
21 |
use_safetensors=True,
|
22 |
custom_pipeline="lpw_stable_diffusion_xl",
|
23 |
+
add_watermarker=False
|
|
|
24 |
)
|
25 |
pipe.to('cuda')
|
26 |
|