Also clear CUDA cache twice if using breakmodel

2021-08-20 11:17:34 -04:00 · 2021-08-20 11:17:34 -04:00 · 5f82e5ba0d
parent f986c65a4e
commit 5f82e5ba0d
1 changed files with 1 additions and 1 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -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)