项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 907
使用教程
环境要求
- Python 3.9+
- GPU(微调时)
安装与启动步骤
-
1装包
PyLate 基于 Sentence Transformers,专门做 ColBERT 类模型。
pip install pylate -
2加载模型
用 PyLate 的包装器加载 ColBERT 权重,输出多向量表示。
from pylate import models m = models.ColBERT(model_name_or_path="lightonai/GTE-ModernColBERT-v1") -
3建索引检索
用 PLAID 索引做快速检索,比逐条打分快几个数量级。
-
4微调
带标注的(query, positive)对即可训练,注意 batch 内负样本的构造。
如何确认成功
检索返回的相关文档明显优于 BM25 基线。
注意事项
- ColBERT 存的是多向量,索引体积比单向量大,容量规划要留余量。
核心亮点
- 实现 ColBERT 类模型,token 级交互检索精度高
- 提供完整训练流程,支持自定义数据与模型配置
- 与 Hugging Face 生态集成,上手门槛低
不足之处
- 项目早期,文档和社区支持待观察
- 对大规模索引的工程优化尚不明确
适用场景
- 学术研究:复现晚期交互模型实验
- 垂直领域语义搜索:法律、医疗文档检索
- 问答系统:结合重排序提升答案召回
替代项目
ColBERT、Jina ColBERT、RAGatouille
上一篇:RAG-Retrieval
下一篇:contrastors
同类项目推荐
stable_diffusion_dreambooth_inpa···
开源
用少量图片微调Stable Diffusion,实现个性化生成与局部修复。
Stable Diffusion Dreambooth Inpainting Finetuning
transformer-tricks
开源
查Transformer加速技巧,少走弯路,直接上手
A collection of tricks and tools to speed up transformer models
KnowledgeCircuits
开源
拆解大模型内部知识电路,精准定位事实存储与推理路径
[NeurIPS 2024] Knowledge Circuits in Pretrained Transformers
Simple-Lora
开源
跟着中文教程,训练你的专属虚拟偶像 LoRA
diffusion lora chinese tutorial,虚拟idol训练中文教程
VeriThinker
开源
让模型学会自我验证,推理更快更准
[NeurIPS 2025] VeriThinker: Learning to Verify Makes Reasoning Model Efficient
dashAI
开源
可视化训练评估部署,一条龙搞定 AI 模型
dashAI: an interactive platform for training, evaluating and deploying AI models
align-anything
开源
一套框架搞定全模态对齐训练,省时省力
Align Anything: Training All-modality Model with Feedback
Kiln
开源
一站式搞定 AI 系统开发:从数据标注到微调评估,全流程可视化协作
Build, Evaluate, and Optimize AI Systems. Includes evals, RAG, agents, fine-tuning, ···
simple-diffusion
开源
用最简代码看懂扩散模型,快速跑通图像生成
A minimal implementation of a denoising diffusion model in PyTorch.
