项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 1386
使用教程
环境要求
- Python 3.9
- conda 环境管理工具
- PyTorch 1.13.1+cu116 及对应 CUDA 11.6
- NVIDIA GPU(案例在 V100 上运行)
- git
安装与启动步骤
-
1克隆仓库
从 GitHub 克隆 KnowLM 项目到本地,并进入项目目录。
git clone https://github.com/zjunlp/KnowLM.git cd KnowLM -
2创建并激活环境
使用 conda 创建 Python 3.9 虚拟环境并激活,环境名 knowlm。
conda create -n knowlm python=3.9 -y conda activate knowlm -
3安装 PyTorch
安装与 CUDA 11.6 对应的 PyTorch 1.13.1。若 CUDA 版本不同请自行调整。
pip install torch==1.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116 -
4安装项目依赖
根据 requirements.txt 安装项目所需 Python 依赖。
pip install -r requirements.txt -
5命令行交互
加载 knowlm-13b-base 模型,进入命令行交互模式;单卡显存不足时可加 --multi_gpu。
python examples/generate_finetune.py --base_model zjunlp/knowlm-13b-base-v1.0 --interactive -
6启动 Web 交互
启动 Pretraining 模型的 Web 交互页面,适合灵活调整任务。
python examples/generate_finetune_web.py --base_model zjunlp/knowlm-13b-base-v1.0 -
7启动 vLLM API
启动 vLLM API 服务,监听 8090 端口,使用多卡张量并行加速。需指定模型路径 data/zhixi-13B。
CUDA_VISIBLE_DEVICES=1,2 python inference/launch_vllm.py --port 8090 --model data/zhixi-13B --use-np-weights --max-num-batched-tokens 8000 --dtype half --tensor-parallel-size 2
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
--base_model | 是 | 指定要加载的基础模型名称或路径 | zjunlp/knowlm-13b-base-v1.0 |
--model | 是 | vLLM 服务加载的模型权重路径 | data/zhixi-13B |
--port | 是 | vLLM API 服务监听的端口 | 8090 |
--multi_gpu | 否 | 单卡显存不足时启用多卡加载 | --multi_gpu |
--tensor-parallel-size | 否 | vLLM 张量并行使用的 GPU 数量 | 2 |
如何确认成功
启动 vLLM 后,用 curl POST 请求 http://127.0.0.1:8090/generate 能返回 generated_text 即成功;Web 交互启动后在浏览器打开页面即可确认。
常见问题
Q:单张 GPU 显存不够怎么办?
A:使用 CUDA_VISIBLE_DEVICES 指定多张 GPU 并添加 --multi_gpu 参数,或通过 --allocate 指定每张卡显存(单位 GB)。
Q:如何做信息抽取任务?
A:启动 knowlm-13b-zhixi 的 LoRA 交互,在 instruction 填指令、input 填待抽取句子;也可运行 generate_lora.py 加 --run_ie_cases。
Q:怎么调整生成结果的多样性?
A:需要多样输出可提高 top_k/top_p 和 temperature;需要聚焦高质量可降低这些值或使用 beam search。
Q:Docker 方式怎么用?
A:docker pull zjunlp/knowlm:v.1 拉取镜像,再用 docker run -it zjunlp/knowlm:v.1 /bin/bash 进入容器。
Q:项目还在开发中吗?
A:是,README 说明项目仍在优化和开发,模型权重会定期更新以支持新功能。
注意事项
- 项目仍在优化开发中,模型权重会定期更新。
- 案例在 V100 上运行,其他设备结果可能有差异,可多次运行或改解码参数。
- 指令微调模型默认用 Web 交互;信息抽取需分 instruction 和 input 填写。
- 缺少 GPU 资源时可用 llama.cpp 进行量化。
核心亮点
- 提供完整的知识注入训练流程,从预训练到微调均支持知识增强
- 内置知识评估基准,可量化模型在知识任务上的表现
- 基于主流框架(如 Transformers)开发,易于集成和扩展
不足之处
- 文档和教程相对简略,上手门槛较高
- 社区规模较小,生态和第三方支持待完善
适用场景
- 构建法律、医疗等垂直领域的专业知识问答助手
- 提升企业私有知识库的检索增强生成(RAG)效果
- 用于学术研究,探索知识注入与模型推理的结合
替代项目
LangChain、LlamaIndex、KEPLER
上一篇:Qwen2.5-Math
下一篇:YuLan-Chat
同类项目推荐
awesome-llm-role-playing-with-pe···
开源
一站式收藏LLM角色扮演资源,快速上手人设对话开发
Awesome-llm-role-playing-with-persona: a curated list of resources for large languag···
moment
开源
开箱即用的时间序列基础模型,零样本预测与微调一把梭
MOMENT: A Family of Open Time-series Foundation Models, ICML'24
AI-Ecosystem
开源
一网打尽AI好项目,快速找到你需要的工具
This is a collection of AI ecosystem, which gathers and organizes various interestin···
Awesome-Domain-LLM
开源
一站式找到垂直领域大模型、数据集和评测基准
收集和梳理垂直领域的开源模型、数据集及评测基准。
llm-resource
开源
一页看全 LLM 全栈好资源,找工具不再大海捞针
LLM全栈优质资源汇总
Awesome-Trajectory-Computing
开源
一网打尽轨迹计算AI资源,找论文代码不迷路
A professional list of Deep Learning and Large (Language) Models (LM, LLM, FM) for T···
Dimple
开源
用离散扩散并行生成文本,让多模态大模型不再逐字挤牙膏
Dimple, the first Discrete Diffusion Multimodal Large Language Model
Awesome-Foundation-Models-in-Med···
开源
一页看懂医学影像AI基础模型,找模型不再大海捞针
A curated list of foundation models for vision and language tasks in medical imaging
nlp_xiaojiang
开源
中文NLP一站式工具箱,开箱即用搞定相似度、NER和数据增强
自然语言处理(nlp),小姜机器人(闲聊检索式chatbot),BERT句向量-相似度(Sentence Si···
