rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Also clear CUDA cache twice if using breakmodel

Browse Source
This commit is contained in:
Gnome Ann
2021-08-20 11:17:34 -04:00
parent f986c65a4e
commit 5f82e5ba0d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -1119,7 +1119,7 @@ def generate(txt, min, max):
genselect(genout)
# Clear CUDA cache again if using GPU
if(vars.hascuda and vars.usegpu):
if(vars.hascuda and (vars.usegpu or vars.breakmodel)):
torch.cuda.empty_cache()
set_aibusy(0)