diff --git a/data_juicer/utils/model_utils.py b/data_juicer/utils/model_utils.py index 108b9cae3..838057a52 100644 --- 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()