提交 60d752bc 作者: littlepanda0716

add torch_gc to clear gpu cache in knowledge_based_chatglm.py

上级 c4b52dda
......@@ -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. 新增清理显存机制。
## 使用方式
### 硬件需求
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论