Commit Graph

1321 Commits

Author SHA1 Message Date
littlepanda0716 c4b52dda72 add torch_gc to clear gpu cache in knowledge_based_chatglm.py 2023-04-07 10:46:02 +08:00
littlepanda0716 5664d1ff62 add torch_gc to clear gpu cache in knowledge_based_chatglm.py 2023-04-07 10:45:44 +08:00
littlepanda0716 3cbc6aa77c update knowledge_based_chatglm.py 2023-04-07 10:11:26 +08:00
littlepanda0716 dfe966ed41 update chatglm_llm.py 2023-04-07 09:58:44 +08:00
littlepanda0716 befcdf6e32 update README.md 2023-04-07 09:48:59 +08:00
littlepanda0716 0d0550525a update README.md 2023-04-07 09:48:59 +08:00
imClumsyPanda 63d900607f
Merge pull request #17 from myml/memory
fix: 修复chatglm模型被复制,显存占用过多
2023-04-07 09:43:49 +08:00
littlepanda0716 51c44e3e0a update chatglm_llm.py 2023-04-07 09:28:45 +08:00
imClumsyPanda 3ab6dbee00
Merge pull request #21 from thomas-yanxin/master
Update requirements.txt
2023-04-05 21:58:38 +08:00
thomas-yanxin b6e603a33c
Update requirements.txt 2023-04-05 21:39:42 +08:00
myml bed03a6ff1 fix: 修复chatglm模型被复制,显存占用过多
model作为类成员会在类实例化时进行一次复制
这导致每询问一个问题显存占用就会翻倍
通过将model改成全局变量修复这个问题
2023-04-05 01:17:26 +08:00
littlepanda0716 1851b4f074 update requirements.txt 2023-04-03 21:07:41 +08:00
littlepanda0716 a8abbff4bb update requirements.txt 2023-04-03 20:37:54 +08:00
littlepanda0716 6762ab442f update function name and requirements.txt 2023-04-03 19:15:08 +08:00
littlepanda0716 ca902e1a7b update function name and requirements.txt 2023-04-03 19:13:34 +08:00
imClumsyPanda f0250eef38
Merge pull request #1 from zx2021/master
修复requirements.txt文件
2023-03-31 22:04:10 +08:00
littlepanda0716 0ddffff3fc update requirements.txt 2023-03-31 22:02:41 +08:00
zx2021 7d59541718
Update requirements.txt
修复了运行pip install -r requirements.txt提示ERROR: Invalid requirement: '"detectron2@git+https://github.com/facebookresearch/detectron2.git@v0.6#egg=detectron2"' (from line 4 of requirements.txt)
Hint: = is not a valid operator. Did you mean == ?的错误
2023-03-31 21:49:16 +08:00
zx2021 2d3ea9830e
Update requirements.txt
修复运行pip install -r requirements.txt提示ERROR: Invalid requirement: '"detectron2@git+https://github.com/facebookresearch/detectron2.git@v0.6#egg=detectron2"' (from line 4 of requirements.txt)
Hint: = is not a valid operator. Did you mean == ?的错误
2023-03-31 21:47:58 +08:00
littlepanda0716 2edb920b24 first commit 2023-03-31 20:23:49 +08:00
littlepanda0716 f17d26addf first commit 2023-03-31 20:09:40 +08:00