Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup memory after model run

Browse Source
This commit is contained in:
Sergey Borisov 2023-09-01 20:50:39 +03:00
parent 1a9f552a75
commit 5151798a16
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/image_util/lama.py
View File

@ -53,5 +53,6 @@ class LaMA:
del model del model
gc.collect() gc.collect()
torch.cuda.empty_cache()
return infilled_image return infilled_image