Update app.py

app.py

@@ -56,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"), dtype=torch.float16)
+    vae = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir, "vae"), torch_dtype=torch.float16)
    
     pipe = StableDiffusionXLPipeline.from_pretrained(
         ckpt_dir,
         vae=vae,
-        dtype=torch.float16,
+        torch_dtype=torch.float16,
         use_safetensors=True,
         #variant="fp16"
     )