From 60d752bc18a5a37325f927ec48905af2a977b926 Mon Sep 17 00:00:00 2001 From: littlepanda0716 Date: Fri, 7 Apr 2023 10:53:03 +0800 Subject: [PATCH] add torch_gc to clear gpu cache in knowledge_based_chatglm.py --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 6958d06..44d2920 100644 --- a/README.md +++ b/README.md @@ -12,8 +12,9 @@ ## 更新信息 -**[2023/04/07]** 解决加载 ChatGLM 模型时发生显存占用为双倍的问题 (感谢 [@suc16](https://github.com/suc16) 和 [@myml](https://github.com/myml)) - +**[2023/04/07]** +1. 解决加载 ChatGLM 模型时发生显存占用为双倍的问题 (感谢 [@suc16](https://github.com/suc16) 和 [@myml](https://github.com/myml)) ; +2. 新增清理显存机制。 ## 使用方式 ### 硬件需求