Update app.py

app.py

@@ -125,7 +125,7 @@ def load_and_prepare_model():
     
 pipe = load_and_prepare_model()
 
-sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/stable-diffusion-xl-refiner-1.0-bf16', subfolder='scheduler',beta_schedule="scaled_linear", token=HF_TOKEN)
+vaeX = AutoencoderKL.from_pretrained('stabilityai/stable-diffusion-xl-refiner-1.0',subfolder='vae', token=HF_TOKEN) #.to(torch.bfloat16)   #  ,use_safetensors=True  FAILS
 
 refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
         "ford442/stable-diffusion-xl-refiner-1.0-bf16",
@@ -134,6 +134,8 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
 )
 
 refiner.vae=vaeX
+refiner.vae.set_default_attn_processor()
+
 refiner.to(device=device)
 
 MAX_SEED = np.iinfo(np.int32).max