RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-01-19 21:37:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'master' into dev

Browse Source
This commit is contained in:
imClumsyPanda 2023-05-03 13:15:00 +08:00
commit 4e9de7df41
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/__init__.py
View File

@ -1,5 +1,4 @@
import torch.cuda
import torch.backends
import torch
def torch_gc(DEVICE):
if torch.cuda.is_available():
@ -8,7 +7,6 @@ def torch_gc(DEVICE):
torch.cuda.ipc_collect()
elif torch.backends.mps.is_available():
try:
import torch.mps
torch.mps.empty_cache()
except Exception as e:
print(e)