Update app.py

app.py

@@ -17,7 +17,7 @@ ckpt = "sdxl_lightning_1step_unet_x0.safetensors"
 unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
 unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device="cuda"))
 pipe_lightning = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
-
+del unet
 pipe_lightning.scheduler = EulerDiscreteScheduler.from_config(pipe_lightning.scheduler.config, timestep_spacing="trailing", prediction_type="sample")
 pipe_lightning.to("cuda")
 
@@ -30,7 +30,7 @@ unet.load_state_dict(load_file(hf_hub_download(repo_name, ckpt_name), device="cu
 pipe_hyper = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
 pipe_hyper.scheduler = LCMScheduler.from_config(pipe_hyper.scheduler.config)
 pipe_hyper.to("cuda")
-
+del unet
 
 @spaces.GPU
 def run_comparison(prompt):