fix: ae

app.py

@@ -118,8 +118,6 @@ class FluxEditor_lore_demo:
         self.info['feature'] = {}
         self.info['inject_step'] = injection_num_steps
         self.info['wh'] = (width, height)
-        
-        torch.cuda.empty_cache()
 
         inp = prepare(self.t5, self.clip, self.init_image, prompt=src_prompt)
         timesteps = get_schedule(inversion_num_steps, inp["img"].shape[1], shift=True)
@@ -196,7 +194,6 @@ class FluxEditor_lore_demo:
         self.model.to(self.device)
         self.t5.to('cpu')
         self.clip.to('cpu')
-        torch.cuda.empty_cache()
         if training_epochs != 0:
             torch.set_grad_enabled(True)
             inp_optim["img"] = self.z0