Update app.py

app.py

@@ -66,6 +66,7 @@ vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", us
 pipe = StableDiffusion3Pipeline.from_pretrained(
     #"stabilityai  #  stable-diffusion-3.5-large",
     "ford442/stable-diffusion-3.5-large-bf16",
+    vae=None,
     #vae=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", use_safetensors=True, subfolder='vae',token=True),
      #scheduler = FlowMatchHeunDiscreteScheduler.from_pretrained('ford442/stable-diffusion-3.5-large-bf16', subfolder='scheduler',token=True),
     # text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True),
@@ -82,7 +83,7 @@ pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safe
 
 pipe.to(device=device, dtype=torch.bfloat16)
 #pipe.to(device)
-pipe.vae=vaeX
+#pipe.vae=vaeX.to('cpu')
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
 
 MAX_SEED = np.iinfo(np.int32).max