Update app.py

app.py

@@ -109,8 +109,8 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
 
 def load_and_prepare_model():
     #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
-    #vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None,use_safetensors=False,token=HF_TOKEN)
-    vaeRV = AutoencoderKL.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='vae', safety_checker=None,token=HF_TOKEN)
+    vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None,use_safetensors=False,token=HF_TOKEN)
+    #vaeRV = AutoencoderKL.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='vae', safety_checker=None,token=HF_TOKEN)
     #vae = AutoencoderKL.from_single_file('https://huggingface.co/ford442/sdxl-vae-bf16/mySLR/myslrVAE_v10.safetensors')
     #vaeX = AutoencoderKL.from_pretrained('ford442/Juggernaut-XI-v11-fp32',subfolder='vae')   #  ,use_safetensors=True  FAILS
     #unetX = UNet2DConditionModel.from_pretrained('ford442/RealVisXL_V5.0_BF16',subfolder='unet').to(torch.bfloat16)   #  ,use_safetensors=True  FAILS
@@ -171,7 +171,7 @@ def load_and_prepare_model():
     #pipe.scheduler = DPMSolverMultistepScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler', algorithm_type='sde-dpmsolver++')
 
     #****  NEVER TO BFLOAT BEFORE CUDA****#
-    pipe.vae = vaeRV #.to(torch.bfloat16)
+    pipe.vae = vaeX #.to(torch.bfloat16)
     #****  NEVER TO BFLOAT BEFORE CUDA****#
 
     #pipe.unet = unetX