We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7567adf commit 0ac1d5bCopy full SHA for 0ac1d5b
1 file changed
examples/dreambooth/train_dreambooth_lora_hidream.py
@@ -236,7 +236,6 @@ def log_validation(
236
}
237
)
238
239
- pipeline.to("cpu")
240
del pipeline
241
free_memory()
242
0 commit comments