Update app.py

app.py

@@ -135,7 +135,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 = vaeX #.to(torch.bfloat16)
+    #pipe.vae = vaeX #.to(torch.bfloat16)
     #****  NEVER TO BFLOAT BEFORE CUDA****#
 
     #pipe.vae.force_upcast=True
@@ -157,7 +157,7 @@ def load_and_prepare_model():
     #**** BETTER WAY ****#
     pipe.to(device, torch.bfloat16)
     #**** BETTER WAY ****#
-    
+    pipe.vae = vaeX #.to(torch.bfloat16)
     #pipe.to(device)
     #pipe.to(torch.bfloat16)