From 3f431994491931686224dd8d65b9e7b72ec72e96 Mon Sep 17 00:00:00 2001
From: imClumsyPanda <littlepanda0716@gmail.com>
Date: Wed, 3 May 2023 13:14:36 +0800
Subject: [PATCH] update utils

---
 utils/__init__.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/utils/__init__.py b/utils/__init__.py
index 37deccf..6cde439 100644
--- a/utils/__init__.py
+++ b/utils/__init__.py
@@ -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)