offload t5 and clip

app.py

@@ -196,6 +196,9 @@ class FluxEditor_lore_demo:
         t0 = time.perf_counter() 
         print(f'optimizing & editing noise, {target_prompt} with seed {seed}, noise_scale {noise_scale}, training_epochs {training_epochs}')
         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
@@ -243,6 +246,8 @@ class FluxEditor_lore_demo:
                 Image.fromarray(binary_mask, mode="L").save(output_path.replace(target_object,f'{target_object}_mask'))
             t1 = time.perf_counter()  
             print(f"Done in {t1 - t0:.1f}s.", f'Saving {output_path} .' if self.save else 'No saving files.')
+            self.t5.to(self.device)
+            self.clip.to(self.device)
         
         return img