Update app.py

app.py

@@ -8,7 +8,7 @@ import uuid
 from PIL import Image
 from huggingface_hub import snapshot_download
 from diffusers import AutoencoderKL
-from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler, AutoPipelineForText2Image, DiffusionPipeline
+from diffusers import StableDiffusionXLPipeline
 from diffusers import EulerAncestralDiscreteScheduler, DPMSolverMultistepScheduler, DPMSolverSDEScheduler
 from diffusers.models.attention_processor import AttnProcessor2_0
 import torch
@@ -16,7 +16,6 @@ from typing import Tuple
 from datetime import datetime
 import requests
 import torch
-from diffusers import DiffusionPipeline
 import importlib
 import re
 from urllib.parse import urlparse
@@ -57,12 +56,12 @@ def load_pipeline_and_scheduler():
     ckpt_dir = snapshot_download(repo_id=cfg["model_id"])
 
     # Load the models
-    vae = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir, "vae"), torch_dtype=torch.float16)
+    vae = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir, "vae"), dtype=torch.float16)
    
     pipe = StableDiffusionXLPipeline.from_pretrained(
         ckpt_dir,
         vae=vae,
-        torch_dtype=torch.float16,
+        dtype=torch.float16,
         use_safetensors=True,
         #variant="fp16"
     )