From 81530133ff341894d69a8b4d0fc105efcb80f5c2 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Wed, 15 Nov 2023 16:22:32 +0800
Subject: [PATCH] fix #1507

Former-commit-id: 1ba9c53bd9743fa95fca1516c0ed9da352dbe9a1
---
 src/cli_demo.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/cli_demo.py b/src/cli_demo.py
index cccda517..781aa551 100644
--- a/src/cli_demo.py
+++ b/src/cli_demo.py
@@ -1,4 +1,5 @@
 from llmtuner import ChatModel
+from llmtuner.extras.misc import torch_gc
 
 try:
     import platform
@@ -27,6 +28,7 @@ def main():
 
         if query.strip() == "clear":
             history = []
+            torch_gc()
             print("History has been removed.")
             continue