Update app.py

app.py

@@ -78,10 +78,10 @@ def load_and_prepare_models():
         #use_safetensors=False,
     )
     torch.cuda.empty_cache()
-    torch.cuda.reset_peak_memory_stats()
     pipe.to(device=device, dtype=torch.bfloat16)
     pipe.vae=vaeX.to(device)
     upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
+    torch.cuda.empty_cache()
     return pipe, upscaler
 
 text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(device=device, dtype=torch.bfloat16)
@@ -92,7 +92,7 @@ pipe, upscaler_2 = load_and_prepare_models()
 
 #pipe.to(device)
 
-upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
+#upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096