weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' into dev

This commit is contained in:
imClumsyPanda 2023-05-03 13:15:00 +08:00
parent 8c4b2f9f6d 3f43199449
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
import torch.backends
def torch_gc(DEVICE): def torch_gc(DEVICE):
if torch.cuda.is_available(): if torch.cuda.is_available():
@ -8,7 +7,6 @@ def torch_gc(DEVICE):
torch.cuda.ipc_collect() torch.cuda.ipc_collect()
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
try: try:
import torch.mps
torch.mps.empty_cache() torch.mps.empty_cache()
except Exception as e: except Exception as e:
print(e) print(e)