Add gc again

2024-02-17 14:53:54 +01:00
parent e1c8e61291
commit 8b7b957b3e
1 changed files with 3 additions and 0 deletions
@@ -1,3 +1,4 @@
+import gc
 import random
 from pathlib import Path
 from time import time
@@ -92,6 +93,8 @@ class Loader:
        self.cache = None
        self.tokenizer = None
        self.generator = None
+
+        gc.collect()
        soft_empty_cache()