Update app.py

app.py

@@ -84,6 +84,7 @@ def load_pipeline(model_name):
         pipe = StableDiffusionXLPipeline.from_pretrained(
             model_name,
             torch_dtype=torch.float16,
+            variant="fp16",
             custom_pipeline="lpw_stable_diffusion_xl",
             safety_checker=None,
             use_safetensors=True,