Closed i0ntempest closed 1 year ago
Oh... just noticed #475 fixes this.
@d8ahazard I've added the required info, mind reopening?
try:
if g_cuda:
del g_cuda
except Exception:
pass
At line 933 This fixes it, although in a quick and ugly way.
try: if g_cuda: del g_cuda except Exception: pass
At line 933 This fixes it, although in a quick and ugly way.
I've done exactly that before I saw #475. Thx still.
Please find the following lines in the console and paste them below. If you do not provide this information, your issue will be automatically closed.
Have you read the Readme? Yes Have you completely restarted the stable-diffusion-webUI, not just reloaded the UI? Yes Have you updated Dreambooth to the latest revision? Yes Have you updated the Stable-Diffusion-WebUI to the latest version? Yes No, really. Please save us both some trouble and update the SD-WebUI and Extension and restart before posting this. Reply 'OK' Below to acknowledge that you did this. OK Describe the bug During the cleanup after saving weights (line 942),
g_cuda
is referenced before assignment in theif
statement, causing training to be interrupted. I wrapped it with atry
and it looks like it does not affect the actual training.Provide logs This log contains an unrelated CUDA OOM error before the actual error I want to report.
Environment
What OS? Windows 11 If Windows - WSL or native? Native What GPU are you using? RTX A2000 12GB Screenshots/Config db_config.json