Update app.py

app.py

@@ -125,6 +125,7 @@ FTP_DIR = os.getenv("FTP_DIR")
 # os.putenv('TORCH_LINALG_PREFER_CUSOLVER','1')
 os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+os.environ["SAFETENSORS_FAST_GPU"] = "1"
 
 upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
 
@@ -204,9 +205,8 @@ def load_and_prepare_model():
     pipe.vae.eval()
     pipe.text_encoder.eval()
     '''
-
     #pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
-    
+    pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/FLUX-dev-lora-add_details.safetensors", low_cpu_mem_usage=False)
     #pipe.unet.to(memory_format=torch.channels_last)
     #pipe.enable_vae_tiling()
     #pipe.unet = torch.compile(pipe.unet, backend="hidet", dynamic=False, mode='max-autotune') #.to(device=device, dtype=torch.bfloat16)