rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2024-12-28 01:00:53 +01:00
Code Issues Projects Releases Wiki Activity

Specifically import torch.cuda.comm in breakmodel.py

Browse Source
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.cuda.comm
import copy
import gc