Skip to content

Commit 0ac1d5b

Browse files
[Hi-Dream LoRA] fix bug in validation (#11439)
remove unnecessary pipeline moving to cpu in validation Co-authored-by: Sayak Paul <[email protected]>
1 parent 7567adf commit 0ac1d5b

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

examples/dreambooth/train_dreambooth_lora_hidream.py

-1
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,6 @@ def log_validation(
236236
}
237237
)
238238

239-
pipeline.to("cpu")
240239
del pipeline
241240
free_memory()
242241

0 commit comments

Comments
 (0)