rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Also clear CUDA cache twice if using breakmodel

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
Show Stats 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) genselect(genout)
# Clear CUDA cache again if using GPU # 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() torch.cuda.empty_cache()
set_aibusy(0) set_aibusy(0)