granite-embedding-small-english-r2 ibm-granite

类型
model
任务
feature-extraction
框架
sentence-transformers
下载量
4,479,791
点赞
75
访问
public
文件
0

标签

  • 文本嵌入
  • 语义搜索
  • 文本相似度
  • 向量检索
  • RAG
  • sentence-transformers
  • NLP
  • 轻量级

摘要

该模型是 IBM Granite Embeddings 系列中的轻量级英文双编码器嵌入模型，仅有47M参数，可生成384维文本向量，支持最高8192 token 的上下文长度。它基于 ModernBERT 架构，采用检索导向预训练、对比微调、知识蒸馏与模型合并等方法训练，在信息检索、代码检索、长文档搜索、多轮对话检索等基准上表现强劲。适用于文本相似度计算、语义搜索、RAG 检索等场景，并支持 sentence-transformers 与 transformers 两种使用方式。

README

--- language: - en library_name: sentence-transformers license: apache-2.0 pipeline_tag: feature-extraction tags: - granite - embeddings - transformers - mteb - feature-extraction --- # Granite-Embedding-Small-English-R2 <!-- 提供模型及其功能的快速摘要。 --> **模型摘要：** Granite-embedding-small-english-r2 是 Granite…

查看完整页面 · 查看原文