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

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
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