From 8b876b103f050252fe2f5417097391a1260e76e2 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Wed, 14 Aug 2024 17:43:19 +0800
Subject: [PATCH] =?UTF-8?q?feat=20=20=20sketch=20=E6=8F=90=E5=8F=96?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3=20fix?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/service/image2sketch/server.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/app/service/image2sketch/server.py b/app/service/image2sketch/server.py
index ebd363e..7a15b55 100644
--- a/app/service/image2sketch/server.py
+++ b/app/service/image2sketch/server.py
@@ -55,6 +55,9 @@ class Image2SketchServer:
         self.data['A'] = transform(A)
         self.data['A'] = self.data['A'].unsqueeze(0).to(device)
 
+    def __del__(self):
+        torch.cuda.empty_cache()
+
     def get_result(self):
         self.model.set_input(self.data)
         self.model.test()  # run inference