fix #1507

Former-commit-id: 2f02f688e15db8bf8f0dad3b7284863998c207a3
2025-08-24 23:02:49 +08:00 · 2023-11-15 16:22:32 +08:00 · 2023-11-15 16:22:32 +08:00 · 8ee48a9c9e
commit 8ee48a9c9e
parent 2ba88c6b08
1 changed files with 2 additions and 0 deletions
--- 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