garbage collection

Author: Bluear7878

tests/flux/test_flux_txt2img_cache_controlnet.py

@@ -14,6 +14,7 @@
 from nunchaku.caching.diffusers_adapters import apply_cache_on_pipe
 from nunchaku.models.transformers.transformer_flux import NunchakuFluxTransformer2dModel
 from nunchaku.utils import get_precision
+import gc
 
 bfl_repo = "black-forest-labs/FLUX.1-dev"
 dtype = torch.bfloat16  # or torch.float16, or torch.float32
@@ -89,6 +90,9 @@
 ).images[0]
 txt2img_res.save("flux.1-dev-txt2img.jpg")
 
+gc.collect()
+torch.cuda.empty_cache()
+
 # cache
 apply_cache_on_pipe(
     pipe_cn,