Unverified Commit 0ac1d5b4 authored by Linoy Tsaban's avatar Linoy Tsaban Committed by GitHub
Browse files

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



remove unnecessary pipeline moving to cpu in validation
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 7567adfc
...@@ -236,7 +236,6 @@ def log_validation( ...@@ -236,7 +236,6 @@ def log_validation(
} }
) )
pipeline.to("cpu")
del pipeline del pipeline
free_memory() free_memory()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment