Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import PIL.Image
 import torch
 import torchvision.transforms.functional as TF
-from diffusers import EulerAncestralDiscreteScheduler, StableDiffusionXLAdapterPipeline, T2IAdapter
+from diffusers import EulerAncestralDiscreteScheduler, StableDiffusionXLAdapterPipeline, T2IAdapter, AutoencoderKL
 
 DESCRIPTION = "# T2I-Adapter-SDXL Sketch"
 
@@ -68,11 +68,11 @@ if torch.cuda.is_available():
     scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
     pipe = StableDiffusionXLAdapterPipeline.from_pretrained(
         model_id,
+        vae=AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16),
         adapter=adapter,
-        safety_checker=None,
+        scheduler=scheduler,
         torch_dtype=torch.float16,
         variant="fp16",
-        scheduler=scheduler,
     )
     pipe.to(device)
 else: