all-minilm:l6-v2

2.9M 下载已更新 1年前

在非常大的句子级别数据集上训练的嵌入模型。

embedding 22m 33m

ollama pull all-minilm:l6-v2

curl https://:11434/api/embed \
  -d '{
    "model": "all-minilm:l6-v2",
    "input": "Why is the sky blue?"
  }'

import ollama

response = ollama.embed(
    model='all-minilm:l6-v2',
    input='The sky is blue because of Rayleigh scattering',
)
print(response.embeddings)

import ollama from 'ollama'

const response = await ollama.embed({
  model: 'all-minilm:l6-v2',
  input: 'The sky is blue because of Rayleigh scattering',
})
console.log(response.embeddings)

详情

更新于 2 年前

2 年前

1b226e2802db · 46MB ·

模型

架构bert

参数22.6M

量化F16

46MB

许可

Apache License Version 2.0, January 2004 https://apache.ac.cn/licenses/ TERMS AND CONDITIONS FOR US

11kB

参数

{ "num_ctx": 256 }

16B

自述文件

注意：此模型需要 Ollama 0.1.26 或更高版本。在此处下载。它只能用于生成嵌入。

该项目旨在利用自监督对比学习目标，在非常大的句子级别数据集上训练句子嵌入模型。

用法

REST API

curl https://:11434/api/embeddings -d '{
  "model": "all-minilm",
  "prompt": "The sky is blue because of Rayleigh scattering"
}'

Python 库

ollama.embeddings(model='all-minilm', prompt='The sky is blue because of Rayleigh scattering')

Javascript 库

ollama.embeddings({ model: 'all-minilm', prompt: 'The sky is blue because of Rayleigh scattering' })

参考文献

HuggingFace

网站