From 0ac1d5b4822e7622da26e7ee11effdb869cd4989 Mon Sep 17 00:00:00 2001 From: Linoy Tsaban <57615435+linoytsaban@users.noreply.github.com> Date: Mon, 28 Apr 2025 19:22:32 +0300 Subject: [PATCH] [Hi-Dream LoRA] fix bug in validation (#11439) remove unnecessary pipeline moving to cpu in validation Co-authored-by: Sayak Paul --- examples/dreambooth/train_dreambooth_lora_hidream.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth_lora_hidream.py b/examples/dreambooth/train_dreambooth_lora_hidream.py index 26a920906b..fbf62999d6 100644 --- a/examples/dreambooth/train_dreambooth_lora_hidream.py +++ b/examples/dreambooth/train_dreambooth_lora_hidream.py @@ -236,7 +236,6 @@ def log_validation( } ) - pipeline.to("cpu") del pipeline free_memory()