AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Hi-Dream LoRA] fix bug in validation (#11439)

Browse Source 
remove unnecessary pipeline moving to cpu in validation

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
Linoy Tsaban
2025-04-28 19:22:32 +03:00
committed by GitHub
parent 7567adfc45
commit 0ac1d5b482
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/dreambooth/train_dreambooth_lora_hidream.py
View File

@@ -236,7 +236,6 @@ def log_validation(
}
)
pipeline.to("cpu")
del pipeline
free_memory()