extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-01-23 14:50:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

When reduced_memory is True, on crash only move model back to Cpu.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-26 16:52:31 -04:00
parent ae40b6ba8c
commit d3df113fb0
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ui/sd_internal/runtime.py
View File

@ -374,9 +374,17 @@ def mk_img(req: Request):
yield from do_mk_img(req)
except Exception as e:
print(traceback.format_exc())
# Model crashed, release all resources in unknown state.
unload_models()
unload_filters()
if thread_data.reduced_memory:
thread_data.modelFS.to('cpu')
thread_data.modelCS.to('cpu')
thread_data.model.model1.to("cpu")
thread_data.model.model2.to("cpu")
else:
# Model crashed, release all resources in unknown state.
unload_models()
unload_filters()
gc() # Release from memory.
yield json.dumps({
"status": 'failed',