From c91d60078f748edcb490abf22f8437ff5a3cfff8 Mon Sep 17 00:00:00 2001 From: linoytsaban Date: Mon, 28 Apr 2025 15:17:39 +0300 Subject: [PATCH] remove unnecessary pipeline moving to cpu in validation --- 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 26a920906b3e..fbf62999d637 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()