fix deprecation of clear_objs_and_retain_memory

LittleNyima · LittleNyima · commit a59ed84b52c7 · 2024-09-30T20:45:53.000+08:00
diff --git a/finetune/train_cogvideox_lora.py b/finetune/train_cogvideox_lora.py
@@ -40,7 +40,7 @@
 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
 

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@`
`40`	`40`	`from diffusers.pipelines.cogvideo.pipeline_cogvideox import get_resize_crop_region_for_grid`
`41`	`41`	`from diffusers.training_utils import (`
`42`	`42`	`cast_training_params,`
`43`		`- clear_objs_and_retain_memory,`
	`43`	`+ free_memory,`
`44`	`44`	`)`
`45`	`45`	`from diffusers.utils import check_min_version, convert_unet_state_dict_to_peft, export_to_video, is_wandb_available`
`46`	`46`	`from diffusers.utils.hub_utils import load_or_create_model_card, populate_model_card`
`@@ -726,7 +726,7 @@ def log_validation(`
`726`	`726`	`}`
`727`	`727`	`)`
`728`	`728`
`729`		`- clear_objs_and_retain_memory([pipe])`
	`729`	`+ free_memory()`
`730`	`730`
`731`	`731`	`return videos`
`732`	`732`