From a59ed84b52c77b88231745cf194ddd216df50560 Mon Sep 17 00:00:00 2001 From: LittleNyima Date: Mon, 30 Sep 2024 20:45:53 +0800 Subject: [PATCH] fix deprecation of clear_objs_and_retain_memory --- finetune/train_cogvideox_lora.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/finetune/train_cogvideox_lora.py b/finetune/train_cogvideox_lora.py index 137f322..a746af8 100644 --- a/finetune/train_cogvideox_lora.py +++ b/finetune/train_cogvideox_lora.py @@ -40,7 +40,7 @@ from diffusers.optimization import get_scheduler from diffusers.pipelines.cogvideo.pipeline_cogvideox import get_resize_crop_region_for_grid from diffusers.training_utils import ( cast_training_params, - clear_objs_and_retain_memory, + free_memory, ) from diffusers.utils import check_min_version, convert_unet_state_dict_to_peft, export_to_video, is_wandb_available from diffusers.utils.hub_utils import load_or_create_model_card, populate_model_card @@ -726,7 +726,7 @@ def log_validation( } ) - clear_objs_and_retain_memory([pipe]) + free_memory() return videos