This commit is contained in:
hiyouga 2023-11-15 16:22:32 +08:00
parent 829e879e04
commit 2f02f688e1
1 changed files with 2 additions and 0 deletions

View File

@ -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