pre-commit done

modelscope · Sep 3, 2024 · 7c2bbb8 · 7c2bbb8
1 parent 379a721
commit 7c2bbb8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/data_juicer/utils/model_utils.py b/data_juicer/utils/model_utils.py
@@ -596,6 +596,4 @@ def free_models():
             MODEL_ZOO[model_key].to('cpu')
         except Exception:
             pass
-    del MODEL_ZOO
-    global MODEL_ZOO
-    MODEL_ZOO = {}
+    MODEL_ZOO.clear()