garbage collection

Author: Bluear7878

tests/flux/test_flux_txt2img_cache_controlnet.py

@@ -1,3 +1,5 @@
+import gc
+
 import torch
 from diffusers import (
     AutoencoderKL,
@@ -89,6 +91,9 @@
 ).images[0]
 txt2img_res.save("flux.1-dev-txt2img.jpg")
 
+gc.collect()
+torch.cuda.empty_cache()
+
 # cache
 apply_cache_on_pipe(
     pipe_cn,