From 8ee48a9c9e638c4ab47a12ba59b1873398ed00a1 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: 2f02f688e15db8bf8f0dad3b7284863998c207a3
---
 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