Former-commit-id: 2f02f688e15db8bf8f0dad3b7284863998c207a3
This commit is contained in:
hiyouga 2023-11-15 16:22:32 +08:00
parent 2ba88c6b08
commit 8ee48a9c9e

View File

@ -1,4 +1,5 @@
from llmtuner import ChatModel from llmtuner import ChatModel
from llmtuner.extras.misc import torch_gc
try: try:
import platform import platform
@ -27,6 +28,7 @@ def main():
if query.strip() == "clear": if query.strip() == "clear":
history = [] history = []
torch_gc()
print("History has been removed.") print("History has been removed.")
continue continue