examples/dreambooth/train_dreambooth_lora.py · 012d08b1bcd74abbc05a9ef163e41c99bf0e6b2e · renzhc / diffusers_dcu

"magic_pdf/vscode:/vscode.git/clone" did not exist on "7e59b4b651feee1e195cdfc628083df20813f9b6"

Enable dreambooth lora finetune example on other devices (#10602) · 012d08b1

jiqing-feng authored Jan 21, 2025



* enable dreambooth_lora on other devices
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* enable xpu
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* check cuda device before empty cache
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix comment
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* import free_memory
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

---------
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

012d08b1

train_dreambooth_lora.py 56.2 KB

Replace train_dreambooth_lora.py