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

Specifically import torch.cuda.comm in breakmodel.py

This commit is contained in:
Gnome Ann 2021-08-20 10:47:54 -04:00
parent b1c13f832a
commit eef0db8dee
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
breakmodel.py
View File

@ -212,6 +212,7 @@ Released under the Apache License 2.0
import torch import torch
import torch.cuda.comm
import copy import copy
import gc import gc