Algorithm Comprehend Heritage
问·道是一款面向广大中小学生开发的私人文言文学习助手,旨在帮助中小学生更好掌握文言文知识。
基于ChatGLM3-6B,以及wenyanwen-chinese-translate-to-ancient 两个模型,使用了Bigdl-llm的INT4量化策略进行压缩部署。
第一代 问·道 能够支持无需gpu的部署,对RAM占用较小,且有着良好的对话质量。
第二代 问·道 增加了UI界面,使用更简洁方便。且修复了文言字词会重复出现的问题,也提升了回复速度。
- ChatGLM3-int4 : 对话大模型
- TranslateModel : 翻译大模型
- crawler.py: 数据爬虫
- vocab.txt: 文言字词数据
- requirements.txt: 环境依赖文件
- bigdl_llm_lowbit.py: 模型压缩 (无需重复运行)
- web.css: 网页参数格式
- test.html: 网页格式
- main.py: 主体运行文件
- socket.js: 启动浏览器的界面
- 问·道.pdf : 项目技术报告
使用 pip 安装依赖:
pip install -r requirements.txt
- 为了保证
torch
的版本正确,请严格按照 官方文档 的说明安装。
可以直接从仓库下载压缩后的模型文件,亦可以从huggingface上获取完成版 而后运行我们的压缩程序进行压缩
python3 bigdl_llm_lowbit.py
node socket.js
即可开始对话学习啦~